Cross-Modal Fine-Tuning: Align then Refine

Junhong Shen, Liam Li, Lucio M. Dery, Corey Staten, Mikhail Khodak +2 more
2/11/2023
cs.LG

No official code found. Best implementation: sjunhongshen/ORCA (75 stars, MIT, last commit 2024-03-06).

Abstract

Fine-tuning large-scale pretrained models has led to tremendous progress in well-studied modalities such as vision and NLP. However, similar gains have not been observed in many other modalities due to a lack of relevant pretrained models. In this work, we propose ORCA, a general cross-modal fine-tuning framework that extends the applicability of a single large-scale pretrained model to diverse modalities. ORCA adapts to a target task via an align-then-refine workflow: given the target input, ORCA first learns an embedding network that aligns the embedded feature distribution with the pretraining modality. The pretrained model is then fine-tuned on the embedded data to exploit the knowledge shared across modalities. Through extensive experiments, we show that ORCA obtains state-of-the-art results on 3 benchmarks containing over 60 datasets from 12 modalities, outperforming a wide range of hand-designed, AutoML, general-purpose, and task-specific methods. We highlight the importance of data alignment via a series of ablation studies and demonstrate ORCA's utility in data-limited regimes.

View on arXivView PDF

Code Implementations(1)

sjunhongshen/ORCACommunity100%
MIT

Official implementation of ORCA proposed in the paper "Cross-Modal Fine-Tuning: Align then Refine"

755Shell, PythonFeb 6, 20232 years agoMIT

Cite this paper

@article{shen2023crossmodal,
  title  = {Cross-Modal Fine-Tuning: Align then Refine},
  author = {Junhong Shen and Liam Li and Lucio M. Dery and Corey Staten and Mikhail Khodak and Graham Neubig and Ameet Talwalkar},
  year   = {2023},
  eprint = {2302.05738},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2302.05738v2},
  journal = {ICML 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2302.05738.svg)](https://codeofpaper.com/paper/2302.05738)

Discussion