Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Internalizing Geometric Law: Learning from Solver Residuals for Precision-Critical Generation

About

Large Language Models frequently hallucinate in precision-critical domains such as technical diagramming and mechanical design, where outputs must satisfy strict geometric constraints. We study open-ended geometric synthesis from natural language: translating free-form descriptions into precise constructions whose entities must simultaneously satisfy dozens of interacting constraints. To make this tractable, we release PyGeoX, a programmable geometric DSL that compiles declarative constraints into a differentiable loss, and PyGeoX-Bench, a stratified suite of 300 problems with per-constraint verifiable rewards. Using PyGeoX as a verifier, we identify a failure mode we call Outlier Gradient Masking: under global-norm rewards (any scheme that aggregates residuals through a single norm, for example, $\exp(-\mathrm{MSE})$), a single outlier constraint can nullify the learning signal across all others. To address this, we propose Saturating Additive Rewards (SAR), which decompose the reward into bounded per-constraint terms, preserving partial progress and ensuring consistent gradients even under severe violations. Against MSE-based rewards, the natural baseline for geometry solvers, SAR improves the hard-tier solving rate by $2.3\times$, and the resulting 8B model is competitive with much larger frontier systems on this benchmark. We release the engine, benchmark, and data at https://github.com/Huawei-AI4Math/PyGeoX.

Rafael Cabral, Pang Zixi, Ziyi Shou, Shen Xin• 2026

Related benchmarks

TaskDatasetResultRank
Geometry Problem SolvingPyGeoX Easy
Accuracy62
5
Geometry Problem SolvingPyGeoX (Hard)
Accuracy41
5
Geometry Problem SolvingPyGeoX Medium
Accuracy50
5
Geometry Problem SolvingPyGeoX-Wild
Accuracy66
5
Showing 4 of 4 rows

Other info

Follow for update