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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| PDE solving | PDEBench Diff.Reac 1D (test) | nRMSE0.0028 | 13 | |
| PDE solving | PDEBench Diff.Sorp (test) | nRMSE0.0016 | 9 | |
| Cross-modal adaptation | NAS-Bench-360 | Darcy (Relative L2)0.0072 | 9 | |
| PDE solving | PDEBench Advection (test) | nRMSE0.0078 | 9 | |
| Diverse Prediction Tasks | NAS-Bench-360 (test) | Darcy Score0.0072 | 9 | |
| PDE solving | PDEBench Darcy (test) | nRMSE0.079 | 8 | |
| PDE solving | PDEBench S.Water (test) | nRMSE0.0054 | 8 | |
| PDE solving | PDEBench Diff.Reac 2D (test) | nRMSE0.817 | 8 | |
| Aggregate Performance Ranking | PDEBench Multiple Tasks | Avg Rank1.25 | 5 | |
| Burgers | PDEBench | nRMSE0.0108 | 5 |