arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2605.09640cs.CVcs.LG

通过强化微调克服视觉连续学习中的灾难性遗忘

Overcoming Catastrophic Forgetting in Visual Continual Learning with Reinforcement Fine-Tuning

The University of Hong Kong(香港大学);The Hong Kong University of Science and Technology(香港科学与技术大学);Hong Kong Generative AI Research and Development Center(香港生成式人工智能研究与开发中心)

Meng Lou, Hanzhong Guo, Linwei Chen, Yizhou Yu

AI总结:

本文提出RaPO方法,通过轨迹级奖励塑造缓解强化微调在连续学习中的遗忘问题,实验证明其在视觉连续学习中有效减少灾难性遗忘。

摘要:

近期研究表明,强化微调(RFT)比监督微调(SFT)更具有抗灾难性遗忘能力。然而,RFT(如GRPO)在挑战性视觉连续学习设置中是否能有效克服遗忘仍存疑问。通过试点研究,我们发现虽然RFT优于SFT,但仍存在显著遗忘。我们发现轨迹级漂移无关性是瓶颈所在:在候选轨迹中获得相同任务奖励的情况下,前任务策略的KL散度差异显著,这与连续任务中的灾难性遗忘强相关。受此启发,我们提出Retention-aware Policy Optimization(RaPO),一种简单有效的RFT方法,通过轨迹级奖励塑造显式缓解遗忘。RaPO包含两个核心组件:(1)Retain Reward将轨迹级分布漂移转换为连续奖励信号,优先强化每个组内的知识保持轨迹;(2)跨任务优势归一化(CTAN),在任务边界维持奖励统计的持续指数移动平均,以稳定连续学习中的优化进程。利用MLLMs的自由文本泛化能力,我们在五个视觉连续学习设置中全面评估RaPO。大量实验表明,RaPO在视觉连续学习中取得领先性能,显著减少灾难性遗忘同时保持强可塑性。到目前为止,本工作代表了首次系统探索RFT在视觉连续学习中的应用,提供了我们希望将启发未来研究的见解。

英文摘要:

Recent studies suggest that Reinforcement Fine-Tuning (RFT) is inherently more resilient to catastrophic forgetting than Supervised Fine-Tuning (SFT). However, whether RFT (e.g., GRPO) can effectively overcome forgetting in challenging visual continual learning settings, such as class-incremental learning (CIL) and domain-incremental learning (DIL), remains an open problem. Through a pilot study, we confirm that while RFT consistently outperforms SFT, it still suffers from non-negligible forgetting. We empirically trace this bottleneck to Trajectory-level Drift Agnosticism: among candidate rollouts achieving identical task rewards, the KL divergence from the preceding-task policy varies substantially, which strongly correlates with catastrophic forgetting across sequential tasks. Motivated by this insight, we propose Retention-aware Policy Optimization (RaPO), a simple yet effective RFT method that explicitly mitigates forgetting through trajectory-level reward shaping. Specifically, RaPO comprises two core components: (1) Retention Reward that converts trajectory-level distribution drift into a continuous reward signal, preferentially reinforcing knowledge-preserving rollouts within each group; (2) Cross-Task Advantage Normalization (CTAN), which maintains a persistent exponential moving average of reward statistics across task boundaries to stabilize the optimization progress during continual learning. Leveraging the free-form textual generalization of MLLMs, we comprehensively evaluate RaPO across five visual continual learning settings. Extensive experiments demonstrate that RaPO achieves leading performance, substantially reducing catastrophic forgetting while preserving strong plasticity. To the best of our knowledge, this work represents the first systematic exploration of RFT in visual continual learning, offering insights that we hope will inspire future research.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑