Papers
Topics
Authors
Recent
Search
2000 character limit reached

BECoTTA: Input-dependent Online Blending of Experts for Continual Test-time Adaptation

Published 13 Feb 2024 in cs.LG and cs.CV | (2402.08712v3)

Abstract: Continual Test Time Adaptation (CTTA) is required to adapt efficiently to continuous unseen domains while retaining previously learned knowledge. However, despite the progress of CTTA, it is still challenging to deploy the model with improved forgetting-adaptation trade-offs and efficiency. In addition, current CTTA scenarios assume only the disjoint situation, even though real-world domains are seamlessly changed. To address these challenges, this paper proposes BECoTTA, an input-dependent and efficient modular framework for CTTA. We propose Mixture-of Domain Low-rank Experts (MoDE) that contains two core components: (i) Domain-Adaptive Routing, which helps to selectively capture the domain adaptive knowledge with multiple domain routers, and (ii) Domain-Expert Synergy Loss to maximize the dependency between each domain and expert. We validate that our method outperforms multiple CTTA scenarios, including disjoint and gradual domain shits, while only requiring ~98% fewer trainable parameters. We also provide analyses of our method, including the construction of experts, the effect of domain-adaptive experts, and visualizations.

Definition Search Book Streamline Icon: https://streamlinehq.com
References (36)
  1. Covariance-aware feature alignment with pre-computed source statistics for test-time adaptation. arXiv preprint arXiv:2204.13263, 2022.
  2. Gradient based sample selection for online continual learning. Advances in neural information processing systems, 32, 2019.
  3. Rainbow memory: Continual learning with a memory of diverse samples. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp.  8218–8227, 2021.
  4. Online continual learning on a contaminated data stream with blurry task boundaries. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp.  9275–9284, 2022.
  5. Albumentations: Fast and flexible image augmentations. Information, 11(2), 2020. ISSN 2078-2489. doi: 10.3390/info11020125. URL https://www.mdpi.com/2078-2489/11/2/125.
  6. Robustnet: Improving domain generalization in urban-scene segmentation via instance selective whitening. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp.  11580–11590, 2021.
  7. Improving test-time adaptation via shift-agnostic weight regularization and nearest source prototypes. In European Conference on Computer Vision, pp.  440–458. Springer, 2022a.
  8. Improving test-time adaptation via shift-agnostic weight regularization and nearest source prototypes. In European Conference on Computer Vision, pp.  440–458. Springer, 2022b.
  9. The cityscapes dataset for semantic urban scene understanding. In Proceedings of the IEEE conference on computer vision and pattern recognition, pp.  3213–3223, 2016.
  10. Switch transformers: Scaling to trillion parameter models with simple and efficient sparsity. The Journal of Machine Learning Research, 23(1):5232–5270, 2022.
  11. Decorate the newcomers: Visual domain prompt for continual test time adaptation. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 37, pp.  7595–7603, 2023.
  12. Back to the source: Diffusion-driven test-time adaptation. arXiv preprint arXiv:2207.03442, 2022a.
  13. Visual prompt tuning for test-time domain adaptation. arXiv preprint arXiv:2210.04831, 2022b.
  14. Tsit: A simple and versatile framework for image-to-image translation. In Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part III 16, pp.  206–222. Springer, 2020.
  15. Cafa: Class-aware feature alignment for test-time adaptation. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp.  19060–19071, 2023.
  16. Online continual learning on class incremental blurry task configuration with anytime inference. arXiv preprint arXiv:2110.10031, 2021.
  17. Towards open-set test-time adaptation utilizing the wisdom of crowds in entropy minimization. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp.  16380–16389, 2023a.
  18. Tta-cope: Test-time adaptation for category-level object pose estimation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp.  21285–21295, 2023b.
  19. Ttn: A domain-shift aware batch normalization in test-time adaptation. arXiv preprint arXiv:2302.05155, 2023.
  20. Ttt++: When does self-supervised test-time training fail or thrive? Advances in Neural Information Processing Systems, 34:21808–21820, 2021.
  21. Evaluating prediction-time batch normalization for robustness under covariate shift. arXiv preprint arXiv:2006.10963, 2020.
  22. The mapillary vistas dataset for semantic understanding of street scenes. In Proceedings of the IEEE international conference on computer vision, pp.  4990–4999, 2017.
  23. Efficient test-time model adaptation without forgetting. In International conference on machine learning, pp.  16888–16905. PMLR, 2022.
  24. Towards stable test-time adaptation in dynamic wild world. arXiv preprint arXiv:2302.12400, 2023.
  25. Playing for data: Ground truth from computer games. In Leibe, B., Matas, J., Sebe, N., and Welling, M. (eds.), European Conference on Computer Vision (ECCV), volume 9906 of LNCS, pp.  102–118. Springer International Publishing, 2016.
  26. The synthia dataset: A large collection of synthetic images for semantic segmentation of urban scenes. In Proceedings of the IEEE conference on computer vision and pattern recognition, pp.  3234–3243, 2016.
  27. Acdc: The adverse conditions dataset with correspondences for semantic driving scene understanding. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp.  10765–10775, 2021.
  28. Outrageously large neural networks: The sparsely-gated mixture-of-experts layer. arXiv preprint arXiv:1701.06538, 2017.
  29. Ecotta: Memory-efficient continual test-time adaptation via self-distilled regularization. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp.  11920–11929, 2023.
  30. Tent: Fully test-time adaptation by entropy minimization. arXiv preprint arXiv:2006.10726, 2020.
  31. Continual test-time domain adaptation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp.  7201–7211, 2022a.
  32. Adamix: Mixture-of-adaptations for parameter-efficient model tuning. arXiv preprint arXiv:2210.17451, 2022b.
  33. Segformer: Simple and efficient design for semantic segmentation with transformers. Advances in Neural Information Processing Systems, 34:12077–12090, 2021.
  34. Bdd100k: A diverse driving dataset for heterogeneous multitask learning. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp.  2636–2645, 2020.
  35. Meta-dmoe: Adapting to domain shift by meta-distillation from mixture-of-experts. Advances in Neural Information Processing Systems, 35:22243–22257, 2022.
  36. Taming sparsely activated transformer with stochastic experts. arXiv preprint arXiv:2110.04260, 2021.
Citations (4)

Summary

  • The paper presents a novel CTTA framework that leverages a Mixture-of-Domain Low-rank Experts module to dynamically adapt to shifting domains.
  • It employs domain-adaptive routing and a domain-expert synergy loss to selectively update parameters, reducing forgetting while enhancing adaptation speed.
  • Experimental results on benchmarks including CGS show significant IoU improvements and up to 98% fewer parameters compared to state-of-the-art baselines.

Review of "BECoTTA: Input-dependent Online Blending of Experts for Continual Test-time Adaptation"

The paper "BECoTTA: Input-dependent Online Blending of Experts for Continual Test-time Adaptation" introduces a novel framework for addressing the challenges encountered in Continual Test-Time Adaptation (CTTA) scenarios. CTTA is a crucial area in machine learning that deals with adapting models to continually changing domains during the test phase while efficiently preserving previously acquired knowledge. Despite the recognition of CTTA's significance, the current body of research has not fully explored the trade-offs between forgetting and adaptation or the model's efficiency. This paper aims to fill these gaps with its innovative approach.

Contributions and Methodology

The authors propose BECoTTA, a framework that provides input-dependent and efficient CTTA by leveraging a Mixture-of-Domain Low-rank Experts (MoDE) module. The MoDE module integrates two pivotal components:

  1. Domain-Adaptive Routing: This component selectively captures domain-adaptive knowledge utilizing multiple domain routers. This process allows models to adapt to current data distributions dynamically and efficiently, minimizing interference and enhancing adaptation speed.
  2. Domain-Expert Synergy Loss: This component is designed to maximize the dependency between specific domains and their corresponding experts, facilitating more effective collaboration and specialization among domain experts.

By focusing on these components, the BECoTTA framework enables a selective update of domain experts, catering to the domain-specific knowledge required for different test scenarios. This modular design enhances both parameter and memory efficiency by implementing sparse parameter updates only when necessary, thereby preserving past knowledge and ensuring rapid adaptability to new domains.

Additionally, the authors introduce the Continual Gradual Shifts (CGS) benchmark, which offers a more realistic CTTA scenario by considering gradual shifts in domains over time. This benchmark presents the opportunity to test models in environments that closer reflect real-world conditions, such as the seamless transition between weather conditions in autonomous driving scenarios.

Experimental Results

The paper presents an extensive evaluation of BECoTTA's performance across several CTTA scenarios, including CDS-Hard, CDS-Easy, and the newly proposed CGS benchmark. In direct comparisons with strong baseline models such as TENT, SAR, and EcoTTA, BECoTTA consistently outperformed across all tested scenarios. Notably, it achieved a significant increase in performance metrics such as mean Intersection over Union (IoU) while utilizing substantially fewer parameters (~98% fewer in some instances compared to state-of-the-art models), which speaks to its efficiency and practical applicability.

Furthermore, the paper explores the potential of BECoTTA in zero-shot domain generalization tasks, demonstrating its ability to generalize well to unseen domains, reinforced by its performance on varied datasets such as BDD-100k, Mapillary, GTAV, and Synthia.

Implications and Future Work

The authors have laid a robust foundation for exploring CTTA with a more application-oriented approach that capitalizes on low-rank experts for efficient adaptation. The promising results underscore the potential for BECoTTA to be deployed in real-world systems where domain shifts are inevitable, particularly in settings constrained by computational resources and memory, like edge devices.

Potential future work could explore the extension of this framework to other machine learning tasks and domains, evaluating the universality of this approach. Furthermore, there is room to investigate more complex scenarios that feature even finer-grained domain transitions or multi-modal domain environments, accentuating BECoTTA's flexibility and robustness in diverse conditions.

Overall, BECoTTA marks a significant stride in CTTA research, providing both a compelling framework and a set of rigorous benchmarks to drive future exploration in this domain.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.

Continue Learning

We haven't generated follow-up questions for this paper yet.

Collections

Sign up for free to add this paper to one or more collections.

Tweets

Sign up for free to view the 5 tweets with 86 likes about this paper.