Skip to content
arXiv· arXiv:2605.08896· Zhuoyun Li, Boxuan Wang, Jinwei Hu, Xiaowei Huang, Yi Dong·Indexed by this site · Originally published

FragileFlow: Constraining Correct yet Fragile Predictions in Foundation Models via Spectral Control

FragileFlow: Spectral Control of Correct-but-Fragile Predictions for Foundation Model Robustness

AI summary

FragileFlow is a plug-and-play regularizer that uses a calibrated margin buffer to identify fragile samples in LLMs and VLMs that are correctly predicted yet whose probability mass has flowed to the wrong class, and organizes them into a class-wise vulnerable-risk matrix. The paper gives the first PAC-Bayes upper bound for this margin-aware error flow, and states that empirical spectral control can conservatively achieve deterministic worst-class robustness under stability conditions. In multiple-choice LLM benchmarks and few-shot CLIP adaptation, its theoretical risk measure outperforms matched baselines, and in most settings it improves worst-class accuracy after perturbation while maintaining clean accuracy.

Read originalarxiv.org