Blow-Up Annealing for Heterogeneous Sharpness
Implementation & benchmark of arXiv:2607.14716 — Different Singular Limits in a Gene Regulatory Network with Multiple Small Parameters
Source paper: Different Singular Limits in a Gene Regulatory Network with Multiple Small Parameters arXiv:2607.14716 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Assign separate sharpness or temperature parameters to two nonlinear subnetworks and anneal them according to a directional chart instead of driving both to their singular limits at the same rate. The optimizer explicitly tracks the ratio of the two scales and changes the schedule when the local Jacobian approaches a stability or bifurcation boundary. This tests whether the order and relative rate of sharpening, rather than only the final activation shape, controls optimization stability and representation quality.
Formulas
Mathematical statement
The paper studies a vector field $F(z;\varepsilon_1,\varepsilon_2)$ with two small parameters and shows that the limit near $(0,0)$ depends on the path taken. A first blow-up writes $(\varepsilon_1,\varepsilon_2)=(r,r\rho)$ in the chart where $\varepsilon_1$ is dominant, or $(\varepsilon_1,\varepsilon_2)=(r\rho,r)$ in the complementary chart, with radial scale $r\geq0$ and directional ratio $\rho\geq0$. For a neural network, let $u$ and $v$ be outputs of two nonlinear modules, let $\varepsilon_1,\varepsilon_2$ control their activation sharpness, and let $J_t=\partial f_\theta/\partial\theta$ be the Jacobian of one training update. The relevant quantitative signature is chart dependence: statistics such as the dominant eigenvalue $\lambda_{\max}(J_t)$, loss curvature, or trajectory variance should differ between $\varepsilon_2/\varepsilon_1\ll1$, $\varepsilon_2/\varepsilon_1\approx1$, and $\varepsilon_2/\varepsilon_1\gg1$. A local discrete-time stability boundary is estimated from $\rho(J_t)=1$, where $\rho(\cdot)$ is spectral radius; for continuous gradient flow the corresponding boundary is $\operatorname{Re}\lambda_{\max}(J_t)=0$. The ratio $\rho=\varepsilon_2/\varepsilon_1$ is the blown-up directional coordinate and $r$ is the overall sharpening scale.
Implementation notes
1. Integration point: choose an MLP or transformer feed-forward block with two gated branches, such as an attention branch and an MLP branch or two parallel MLP experts. Replace each sigmoid or GELU gate by a temperature-controlled sigmoid $H(z;\varepsilon,\tau)$, using independent temperatures $\varepsilon_1$ and $\varepsilon_2$. Parameterize them as $\varepsilon_1=r$ and $\varepsilon_2=r\rho$ in the first chart; use the swapped chart when $\rho>1$ if numerical conditioning becomes poor. 2. Pseudocode: initialize $r=r_0$ and $\rho=1$; every $K$ optimization steps estimate the largest eigenvalue magnitude of the update Jacobian using one or two Hessian-vector products and power iteration. If $\rho(J_t)>1-\delta$, stop decreasing $r$, reduce the learning rate by $0.5$, and move $\rho$ toward the less-sharp branch; otherwise set $r\leftarrow qr$ with $q\in[0.95,0.999]$. Log loss, gradient norm, $\rho(J_t)$, activation saturation, and the ratio $\rho$. 3. Computed from the paper: the directional blow-up construction and the prediction that different relative scalings have different limiting dynamics. Estimated empirically: the neural-network stability boundary, since exact bifurcation values depend on architecture, data, and optimizer. 4. First cheap experiment: train a two-branch three-layer MLP on MNIST or CIFAR-10 with SGD, comparing joint annealing $(\varepsilon_1,\varepsilon_2)=(r,r)$ against fixed-ratio schedules $\rho\in\{0.1,0.3,1,3,10\}$. Use identical final temperatures and learning-rate budgets. The falsifiable prediction is a sharp change in loss oscillation, gradient variance, or saturation when $\rho$ crosses an architecture-dependent interval; the stable schedule should maintain $\rho(J_t)<1$ and permit a smaller final $r$ without divergence. A quantitative target is that the measured transition in the dominant Jacobian eigenvalue occurs consistently within 20% across random seeds and at least two widths.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.