cs.LGJun 2, 2026

MAdam: Metric-Aware Multi-Objective Adam

Authors: Fengbei LiuRachit SalujaSunwoo KwakRuibo WangRuining DengHeejong KimJohannes C. PaetzoldMert R. Sabuncu

Organizations: Cornell Tech · Delft University of Technology · Weill Cornell Medicine

Abstract

Multi-objective optimization (MOO) underlies many machine learning problems, yet MOO solvers across the loss-balancing, gradient-balancing, and Pareto-based families almost universally hand their reconciled directions to Adam~\cite{kingma2015adam}. We show this coupling introduces two systematic gaps between the solver's intent and the optimizer's execution. The first is a \emph{weighting mismatch}: Adam's second-moment denominator entangles the time-varying preference vector with gradient statistics, marginalizing the preference into a history average and collapsing distinct Pareto trade-offs toward a near-uniform mixture. The second is a \emph{geometric mismatch}: Adam's adaptive metric distorts the Euclidean geometry MOO solvers assume, turning aligned objectives into apparent conflicts. To resolve both jointly, we introduce \textbf{MAdam} (Metric-Aware Multi-Objective Adam), a drop-in wrapper that leaves both solver and optimizer unchanged. MAdam preconditions the reconciled direction by the preference-conditioned curvature of the scalarized objective; on this whitened input, Adam's second moment collapses to identity, so the realized update is governed by the preference-conditioned metric. Across multi-task learning, Pareto-front recovery, physics-informed neural networks, and medical imaging, MAdam consistently improves over Adam for every solver family.

Explore similar work

CardsList
  1. Optimistic Dual Averaging Unifies Modern Optimizers

    May 11, 2026Thomas Pethick, Wanyun Xie, Roman Machacek +1Optimizer DesignFederated Averaging