Task-aware Adaptive Learning for Cross-domain Few-shot Learning

Yurong Guo, Ruoyi Du, Yuan Dong, Timothy Hospedales, Yi-Zhe Song, Zhanyu Ma*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract / Description of output

Although existing few-shot learning works yield promising results for in-domain queries, they still suffer from weak cross-domain generalization. Limited support data requires effective knowledge transfer, but domain-shift makes this harder. Towards this emerging challenge, researchers improved adaptation by introducing task-specific parameters, which are directly optimized and estimated for each task. However, adding a fixed number of additional parameters fails to consider the diverse domain shifts between target tasks and the source domain, limiting efficacy. In this paper, we first observe the dependence of task-specific parameter configuration on the target task. Abundant task-specific parameters may over-fit, and insufficient task-specific parameters may result in under-adaptation -- but the optimal task-specific configuration varies for different test tasks. Based on these findings, we propose the Task-aware Adaptive Network (TA2-Net), which is trained by reinforcement learning to adaptively estimate the optimal task-specific parameter configuration for each test task. It learns, for example, that tasks with significant domain shift usually have a larger need for task-specific parameters for adaptation. We evaluate our model on Meta-dataset. Empirical results show that our model outperforms existing state-of-the-art methods.
Original languageEnglish
Title of host publicationProceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)
Number of pages9
ISBN (Electronic)979-8-3503-0718-4
ISBN (Print)979-8-3503-0719-1
Publication statusPublished - 15 Jan 2024
EventInternational Conference on Computer Vision 2023 - Paris, France
Duration: 2 Oct 20236 Oct 2023

Publication series

NameInternational Conference on Computer Vision (ICCV)
ISSN (Print)1550-5499
ISSN (Electronic)2380-7504


ConferenceInternational Conference on Computer Vision 2023
Abbreviated titleICCV 2023
Internet address


Dive into the research topics of 'Task-aware Adaptive Learning for Cross-domain Few-shot Learning'. Together they form a unique fingerprint.

Cite this