Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Rethinking Cross-Modal Fine-Tuning: Optimizing the Interaction between Feature Alignment and Target Fitting

About

Adapting pre-trained models to unseen feature modalities has become increasingly important due to the growing need for cross-disciplinary knowledge integration. A key challenge here is how to align the representation of new modalities with the most relevant parts of the pre-trained model's representation space to enable accurate knowledge transfer. This requires combining feature alignment with target fine-tuning, but uncalibrated combinations can exacerbate misalignment between the source and target feature-label structures and reduce target generalization. Existing work, however, lacks a theoretical understanding of this critical interaction between feature alignment and target fitting. To bridge this gap, we develop a principled framework that establishes a provable generalization bound on the target error, which explains the interaction between feature alignment and target fitting through a novel concept of feature-label distortion. This bound offers actionable insights into how this interaction should be optimized for practical algorithm design. The resulting approach achieves significantly improved performance over state-of-the-art methods across a wide range of benchmark datasets.

Trong Khiem Tran, Manh Cuong Dao, Phi Le Nguyen, Thao Nguyen Truong, Trong Nghia Hoang• 2026

Related benchmarks

TaskDatasetResultRank
PDE solvingPDEBench Diff.Reac 1D (test)
nRMSE0.0028
13
PDE solvingPDEBench Diff.Sorp (test)
nRMSE0.0016
9
Cross-modal adaptationNAS-Bench-360
Darcy (Relative L2)0.0072
9
PDE solvingPDEBench Advection (test)
nRMSE0.0078
9
Diverse Prediction TasksNAS-Bench-360 (test)
Darcy Score0.0072
9
PDE solvingPDEBench Darcy (test)
nRMSE0.079
8
PDE solvingPDEBench S.Water (test)
nRMSE0.0054
8
PDE solvingPDEBench Diff.Reac 2D (test)
nRMSE0.817
8
Aggregate Performance RankingPDEBench Multiple Tasks
Avg Rank1.25
5
BurgersPDEBench
nRMSE0.0108
5
Showing 10 of 28 rows

Other info

Follow for update