用强化学习提升视觉定位模型在密集场景下的精度与效率。
Falcon Perception-HD: High Density Perception via Reinforcement Learning

- 采用强化学习直接优化感知指标,替代传统最大似然训练。
- 在每场景500个物体的极密场景中达到顶尖性能,消除重复掩码和去重需求。
- 无需负样本训练即可保持对象存在性判断能力,适合高密度视觉理解任务。
自回归感知模型在开放词汇设置下通常采用监督微调(SFT)与最大似然训练,但其优化目标(逐词交叉熵)与实际感知指标(如精确率、召回率)严重不一致。本文探索后训练强化学习(RL),特别是GRPO,以直接对齐模型与评估指标。基于新提出的Falcon Perception,我们设计了针对感知任务特性的RL框架:处理集合结构输出的奖励设计与多头采样控制。实验发现,强化学习带来多重优势:首先,在每场景最多500个物体的极密场景中实现当前最优表现,而多数现有系统在此场景会严重退化或崩溃;其次,有效解决自回归模型常见的掩码重复问题,并几乎完全消除对NMS和坐标去重的需求,提升性能与效率,同时免去超参数调优;整体上,我们在PBench和SACO-Gold数据集上对所有难度级别的指代表达分割任务均有提升;此外,我们提出一种优雅方法,在未训练负样本的情况下保留对象存在性判断能力(通过MCC评估)。简单惩罚误检与漏检的奖励函数已足够有效。我们开发了两种针对性的混合自标注流水线,分别用于困难指代表达和极密场景,显著提升RL训练效果。模型权重已发布为Falcon Perception修订版(https://huggingface.co/tiiuae/Falcon-Perception),数据集将随后公开。
原文摘要 · Abstract (English)
Autoregressive perception models trained to localize visual entities under the open-vocabulary setting are mostly trained using Supervised fine-tuning (SFT) with maximum likelihood, yet it optimizes a proxy objective (per-token cross-entropy) that is fundamentally misaligned with perception metrics such as precision and recall. In this paper, we explore post-training reinforcement learning (RL), specifically GRPO, to directly align these models with their evaluation metrics. Building up on the recently introduced Falcon Perception, we design an RL framework that addresses perception-specific challenges: reward design for set-structured outputs and multi-head sampling control. We discover multiple benefits from RL for perception: first, RL unlocks state-of-the-art performance in very dense scenes (up to 500 objects per scene), a regime where most existing systems degrade sharply or collapse; furthermore it fixes common issues in autoregressive perception models like mask repetitions and removes almost entirely the need for NMS and coordinate deduplication, which improve both performance and efficiency and remove the need for hyperparameters tuning; overall, we notice improvements on all levels of difficulties in referring expression segmentation (on PBench and SACO-Gold), and we find an elegant way to preserve the knowledge of whether an object exists or not (as evaluated by MCC) without training on negative samples. We show that a simple reward that penalizes false negatives and positives is sufficient. We develop two hybrid self-annotation pipelines, respectively tailored for difficult referring expressions and very dense scenes, and show their benefits on RL-training. Model weights are released as a Falcon Perception revision~\footnote{https://huggingface.co/tiiuae/Falcon-Perception}. Datasets will be published.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。