Polyconvex rotation-frame Jacobian loss
Implementation & benchmark of arXiv:2608.23072 — Polyconvexity for Cosserat nonlinear elasticity and nonlinear couple-stress theory
Source paper: Polyconvexity for Cosserat nonlinear elasticity and nonlinear couple-stress theory arXiv:2608.23072 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace ordinary Jacobian penalties in coordinate MLPs or deformation networks with a learned local rotation frame and a polyconvex energy of the relative stretch. Penalize \(U\), its cofactor, and its determinant through a convex function, while separately smoothing the rotation field through \(R^T\operatorname{Curl}R\). The intended benefit is resistance to fold formation and better conditioning than directly penalizing \(\|J-I\|^2\), especially for large deformations.
Formulas
Mathematical statement
The paper lifts the deformation gradient \(D\varphi(x)\in\mathbb R^{3\times3}\) using an independent rotation \(\overline R(x)\in\mathrm{SO}(3)\) and relative stretch \(\overline U(x)=\overline R(x)^T D\varphi(x)\). The key identities are \(\operatorname{Cof}(\overline R^T D\varphi)=\overline R^T\operatorname{Cof}(D\varphi)\) and \(\det(\overline R^T D\varphi)=\det(D\varphi)\), since \(\det\overline R=1\). A polyconvex energy has the form \(W(\overline U,\operatorname{Cof}\overline U,\det\overline U)\), where \(W\) is convex in its three displayed arguments and grows coercively in them. The curvature variable \(K=\overline R^T\operatorname{Curl}\overline R\) controls first-order variation of the rotation field. In the constrained model, \(\overline U\in\operatorname{Sym}^+(3)\) together with \(\det D\varphi>0\) forces \(\overline R=\operatorname{polar}(D\varphi)\). For neural use, \(J=D\varphi\), \(R\) is a predicted or differentiably estimated rotation, \(U=R^TJ\), and the loss is convex in \((U,\operatorname{Cof}U,\det U)\), with a barrier for positive determinant.
Implementation notes
(1) Integrate this into a coordinate MLP that predicts a deformation \(\varphi_\theta(x)\in\mathbb R^3\), either for an implicit shape deformation or for a neural warp used in registration. At sampled spatial points \(x_b\), compute the Jacobian \(J_b=\partial\varphi_\theta/\partial x\) with forward-mode autodiff. Add a second head producing a 9-vector reshaped into a matrix \(A_b\), and project it to \(R_b\in\mathrm{SO}(3)\) by SVD polar projection \(R_b=UV^T\), flipping the last column of \(V\) if \(\det(UV^T)<0\). Compute \(U_b=R_b^TJ_b\), \(C_b=\operatorname{Cof}(U_b)\), and \(d_b=\det(J_b)\). (2) Use this pseudocode: J=jacobian(phi(x)); A=rotation_head(x); R=svd_polar(A); U=transpose(R)@J; C=cofactor(U); d=det(J); loss_data=task_loss(phi(x)); loss_pc=convex_energy(U,C,d); loss_det=-lambda_det*log(d+eps); K=R^T@curl(R); loss=loss_data+lambda_pc*loss_pc+lambda_K*mean(K*K)+loss_det. Estimate \(K\) with centered finite differences of the rotation head at neighboring points; if the application has no spatial coordinates, omit this term. (3) Implement \(W_\theta\) as a small input-convex neural network whose input is the concatenated entries of \((U,C,d)\), with nonnegative hidden-to-hidden weights, or initially use explicit \(a\|U-I\|_F^2+b\|C-I\|_F^2+c(d-1)^2\). The paper supplies the structural variables and rotation-frame identities; autodiff estimates \(J\), finite differences estimate \(\operatorname{Curl}R\), and the convex-energy parameters are learned jointly. Log the fraction of points with \(d\le0\), the Jacobian condition number, and the task loss. (4) First experiment: train a 4-layer SIREN or ReLU coordinate MLP on a synthetic 3D registration task involving a twist plus radial compression, comparing ordinary \(\|J-I\|_F^2\), determinant-only barriers, and this loss at matched parameter count and gradient evaluations. Success is a lower fold rate and lower Jacobian condition number at equal reconstruction error, or faster convergence to the same error. Also test robustness under twice the deformation amplitude; the hypothesis is that the polyconvex minor terms retain accuracy where the baseline develops folds.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 7/10
Built a reproducible PyTorch MVP with SO(3) rotation frames, autodiff Jacobians, stable cofactor computation, determinant barrier, and lifted polyconvex energy. The rotation/cofactor/determinant identities were verified to about 1.4e-6 absolute error. In the toy deformation experiment both methods had zero folds, while the proposed loss improved mean Jacobian conditioning and minimum determinant at both amplitudes; it slightly improved task MSE at amplitude 1 and was nearly equal at amplitude 2. This is a promising conditioning/stability signal, but not definitive evidence of lower fold rate because the tiny task produced no folds for either method.
- Agent confidence
- 7/10
- Baseline
- amp1: MSE 0.005764, fold fraction 0, mean condition 2.553, min det 0.241; amp2: MSE 0.019627, fold fraction 0, mean condition 4.863, min det 0.0674
- Idea
- amp1: MSE 0.004731, fold fraction 0, mean condition 2.380, min det 0.568; amp2: MSE 0.020299, fold fraction 0, mean condition 3.796, min det 0.177
Limitations: Only a small fixed-collocation synthetic warp and one random seed were tested. Rotation smoothness K=R^T Curl R, learned convex energy parameters, larger networks, parameter/FLOP matching, and genuinely folding or noisy registration tasks were not tested; the implementation uses an exponential-map SO(3) head rather than SVD polar projection for numerical stability.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.23072)
Significantly WORSE than the tuned baseline (n/a, p=0.0321, wins 1 / 8 paired seeds).
- Benchmark
- Deformation_registration
- Model
- coordinate_warp
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- 0.0074 (n/a; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0321 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±5.6%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 0 configs
The registered track is a synthetic deformation-registration task rather than a real clinical or graphics dataset. Rotation curl smoothing, learned input-convex energy parameters, larger models, and longer training were not tested. CUDA encountered a native illegal-instruction crash, so the completed protocol ran on CPU.
How to run: python3 run_bench.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
The registered deformation_registration benchmark confirmed the predicted stability mechanism: the polyconvex rotation-frame loss reduced mean Jacobian condition number from 9.6255 to 3.6457, reduced fold fraction from 5.5% to 0%, and improved mean minimum determinant from -0.0314 to 0.1291. However, test MSE worsened by 0.00736 with permutation p=0.0321, yielding the required verdict idea worse (significant), so this is not a benchmark win.
- Baseline mean test mse
- 0.136754
- Delta mean
- 0.007355469278991222
- Idea mean test mse
- 0.144109
- Math check
- map[cofactor_max_abs:7.152557373046875e-07 det_max_abs:4.76837158203125e-07 rotation_det_max_abs:2.384185791015625e-07]
- Mechanism signature
- map[baseline_fold_fraction:0.05500000016763806 baseline_mean_condition:9.625483721494675 baseline_mean_min_det:-0.031383185618324205 confirmed:true idea_fold_fraction:0 idea_mean_condition:3.6457439363002777 idea_mean_min_det:0.1291193552315235]
- Permutation p value
- 0.0321
Benchmark result
Significantly WORSE than the tuned baseline (n/a, p=0.0321, wins 1 / 8 paired seeds).
- Benchmark
- Deformation_registration
- Model
- coordinate_warp
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- 0.0074 (n/a; negative = idea better)
- Wins
- 1 / 8 paired seeds
- p-value
- 0.0321 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±5.6%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 0 configs
The registered track is a synthetic deformation-registration task rather than a real clinical or graphics dataset. Rotation curl smoothing, learned input-convex energy parameters, larger models, and longer training were not tested. CUDA encountered a native illegal-instruction crash, so the completed protocol ran on CPU.
How to run: python3 run_bench.py
Artifacts
- bench_results.json 12.7 KB View Raw
- custom_deformation_track.py 0.9 KB View Raw
- experiment.py 6.5 KB View Raw
- report.md 1.6 KB View
- report_bench_2026-09-03T012737.md 5.7 KB View
- results.json 1.6 KB View Raw
- run_bench.py 5.8 KB View Raw