amazon.scienceTheodore VasiloudisProblem & solution
How controllers from industrial machinery can coordinate multitask machine learning
Summary
Multitask machine learning often suffers from gradient conflicts, drift, and drought when blending multiple objective gradients in single update steps. To address this in graph self-supervised learning, researchers developed ControlG, a framework that allocates discrete computational blocks to individual objectives over time. ControlG operates across three coupled loops: sensing objective difficulty via spectral demand and gradient interference, planning target compute allocations using log-hypervolume Pareto sensitivities, and executing the schedule via a proportional-integral-derivative controller. Evaluated across homophilic and heterophilic graph benchmarks, the approach outperforms traditional multitask gradient-manipulation baselines while incurring low computational overhead. The framework also produces interpretable training timelines and deficit traces.
Context
In multitask machine learning and graph self-supervised learning, blending multiple objective gradients at each training step causes parameter updates to compromise, leading to gradient disagreement, temporal drift, and objective starvation.
Approach / What changed
ControlG treats multitask learning as a temporal scheduling problem by dedicating distinct training blocks to individual objectives. It uses a slow sensing loop to measure spectral demand and gradient interference, an epoch-level planning loop based on Pareto log-hypervolume sensitivities, and a block-level PID controller to track planned allocations.
Takeaways
Simply separating training objectives across time via random block scheduling beats several gradient-manipulation methods like AutoSSL, PCGrad, and CAGrad on node clustering.
ControlG uses spectral demand (measured via the Rayleigh quotient of per-node gradients) and multiple-gradient descent interference to estimate per-objective difficulty.
On the ogbn-arxiv benchmark, ControlG achieved 72.86% node classification accuracy, outperforming CAGrad while running substantially faster than AutoSSL and ParetoGNN.
Source and freshness
- Source
- amazon.science
- Published
- Added to Yomu
- Last updated
Summaries are generated from the original by a language model. Check the original before relying on it.
Keep reading
Related reading
The Good Stuff
Paid picks we use ourselves. Links carry a referral code — a subscription through one earns us a commission at no extra cost to you.
The best engineering reads, once a week
One useful email. Unsubscribe anytime.