MoETTA: Test-Time Adaptation Under Mixed Distribution Shifts with MoE-LayerNorm

TL;DR

MoETTA introduces a novel entropy-based test-time adaptation framework using Mixture-of-Experts to handle complex mixed distribution shifts, outperforming existing methods on new realistic benchmarks.

Contribution

The paper proposes MoETTA, a Mixture-of-Experts based TTA method that enables diverse adaptation directions, addressing limitations of unified adaptation paths under heterogeneous shifts.

Findings

MoETTA achieves state-of-the-art performance on new benchmarks.

Modeling multiple adaptation directions improves robustness.

Experiments demonstrate consistent outperformance over baselines.

Abstract

Test-Time adaptation (TTA) has proven effective in mitigating performance drops under single-domain distribution shifts by updating model parameters during inference. However, real-world deployments often involve mixed distribution shifts, where test samples are affected by diverse and potentially conflicting domain factors, posing significant challenges even for SOTA TTA methods. A key limitation in existing approaches is their reliance on a unified adaptation path, which fails to account for the fact that optimal gradient directions can vary significantly across different domains. Moreover, current benchmarks focus only on synthetic or homogeneous shifts, failing to capture the complexity of real-world heterogeneous mixed distribution shifts. To address this, we propose MoETTA, a novel entropy-based TTA framework that integrates the Mixture-of-Experts (MoE) architecture. Rather than enforcing a single parameter update rule for all test samples, MoETTA introduces a set of structurally decoupled experts, enabling adaptation along diverse gradient directions. This design allows the model to better accommodate heterogeneous shifts through flexible and disentangled parameter updates. To simulate realistic deployment conditions, we introduce two new benchmarks: potpourri and potpourri+. While classical settings focus solely on synthetic corruptions, potpourri encompasses a broader range of domain shifts--including natural, artistic, and adversarial distortions--capturing more realistic deployment challenges. Additionally, potpourri+ further includes source-domain samples to evaluate robustness against catastrophic forgetting. Extensive experiments across three mixed distribution shifts settings show that MoETTA consistently outperforms strong baselines, establishing SOTA performance and highlighting the benefit of modeling multiple adaptation directions via expert-level diversity.