12–17 Jul 2026
University of Graz
Europe/Vienna timezone

Conflict-Gated Gradient Scaling: Resolving gradient pathology in physics-informed epidemiological models

14 Jul 2026, 17:00
20m
11.33 - SR (University of Graz)

11.33 - SR

University of Graz

34
Contributed Talk Cutting Edge Research Areas Contributed Talks

Speaker

Woldegebriel Assefa Woldegerima (York University, Canada)

Description

Physics-Informed Neural Networks (PINNs) are increasingly used in mathematical epidemiology to bridge the gap between noisy clinical data and compartmental models. However, training these hybrid networks is often unstable due to competing optimization objectives. As established in recent literature on “Gradient Pathology,” the gradient vectors derived from the data loss and the physical residual often point in conflicting directions, leading to slow convergence or optimization deadlock. While existing methods attempt to resolve this by balancing gradient magnitudes or projecting conflicting vectors, we propose an efficient alternative: Conflict-Gated Gradient Scaling (CGGS). This method utilizes the cosine similarity between the data and physics gradients to dynamically modulate the penalty weight. Unlike standard annealing schemes that only normalize scales, CGGS acts as a geometric gate: it suppresses the physical constraint when directional conflict is high, allowing the optimizer to prioritize data fidelity, and restores the constraint when gradients align. We prove that this gating mechanism preserves the standard $O(1/T)$ convergence rate for smooth non-convex objectives, a guarantee that fails under fixed-weight or magnitude–balanced training when gradients conflict. We demonstrate that this mechanism autonomously induces a curriculum learning effect, improving parameter estimation in stiff epidemiological systems compared to magnitude-based baselines.

Author

Nickson Golooba (York University, Canada)

Co-author

Woldegebriel Assefa Woldegerima (York University, Canada)

Presentation materials

There are no materials yet.