Hierarchical Adaptive networks with Task vectors for Test-Time Adaptation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ambekar, Sameer, Hasny, Marta, Daza, Laura, Lang, Daniel M., Schnabel, Julia A.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915891357155328
author Ambekar, Sameer
Hasny, Marta
Daza, Laura
Lang, Daniel M.
Schnabel, Julia A.
author_facet Ambekar, Sameer
Hasny, Marta
Daza, Laura
Lang, Daniel M.
Schnabel, Julia A.
contents Test-time adaptation allows pretrained models to adjust to incoming data streams, addressing distribution shifts between source and target domains. However, standard methods rely on single-dimensional linear classification layers, which often fail to handle diverse and complex shifts. We propose Hierarchical Adaptive Networks with Task Vectors (Hi-Vec), which leverages multiple layers of increasing size for dynamic test-time adaptation. By decomposing the encoder's representation space into such hierarchically organized layers, Hi-Vec, in a plug-and-play manner, allows existing methods to adapt to shifts of varying complexity. Our contributions are threefold: First, we propose dynamic layer selection for automatic identification of the optimal layer for adaptation to each test batch. Second, we propose a mechanism that merges weights from the dynamic layer to other layers, ensuring all layers receive target information. Third, we propose linear layer agreement that acts as a gating function, preventing erroneous fine-tuning by adaptation on noisy batches. We rigorously evaluate the performance of Hi-Vec in challenging scenarios and on multiple target datasets, proving its strong capability to advance state-of-the-art methods. Our results show that Hi-Vec improves robustness, addresses uncertainty, and handles limited batch sizes and increased outlier rates.
format Preprint
id arxiv_https___arxiv_org_abs_2508_09223
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Hierarchical Adaptive networks with Task vectors for Test-Time Adaptation
Ambekar, Sameer
Hasny, Marta
Daza, Laura
Lang, Daniel M.
Schnabel, Julia A.
Machine Learning
Artificial Intelligence
Test-time adaptation allows pretrained models to adjust to incoming data streams, addressing distribution shifts between source and target domains. However, standard methods rely on single-dimensional linear classification layers, which often fail to handle diverse and complex shifts. We propose Hierarchical Adaptive Networks with Task Vectors (Hi-Vec), which leverages multiple layers of increasing size for dynamic test-time adaptation. By decomposing the encoder's representation space into such hierarchically organized layers, Hi-Vec, in a plug-and-play manner, allows existing methods to adapt to shifts of varying complexity. Our contributions are threefold: First, we propose dynamic layer selection for automatic identification of the optimal layer for adaptation to each test batch. Second, we propose a mechanism that merges weights from the dynamic layer to other layers, ensuring all layers receive target information. Third, we propose linear layer agreement that acts as a gating function, preventing erroneous fine-tuning by adaptation on noisy batches. We rigorously evaluate the performance of Hi-Vec in challenging scenarios and on multiple target datasets, proving its strong capability to advance state-of-the-art methods. Our results show that Hi-Vec improves robustness, addresses uncertainty, and handles limited batch sizes and increased outlier rates.
title Hierarchical Adaptive networks with Task vectors for Test-Time Adaptation
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2508.09223