用大模型适配器实现实时物体使用属性检测
YOLOA: Real-Time Affordance Detection via LLM Adapter
- 通过轻量级检测器与大模型适配器联合优化物体识别与使用方式
- 在ADG-Det和IIT-Heat上分别达52.8/73.1 mAP,最快达89.77 FPS
- 适合需要实时理解物体功能的机器人、自动驾驶场景
姿态检测旨在通过理解物体的‘是什么’、‘在哪里’和‘如何使用’来解决具身智能中的基本‘何-何-何’挑战。然而,多数方法仅关注物体的‘如何使用’,忽略‘是什么’和‘在哪里’;另一些方法将物体检测与属性学习视为独立任务,缺乏有效交互且难以实现实时处理。为此,我们提出YOLOA,一种基于大语言模型(LLM)适配器的实时属性检测模型,通过统一框架联合处理这两项任务。YOLOA采用轻量级检测器,包含物体检测与属性学习分支,并通过LLM适配器进行精炼。训练时,适配器与初步预测交互,生成更准确的类别先验、框偏移与属性门控信号,协同优化两个分支。在重新标注的ADG-Det和IIT-Heat数据集上的实验表明,YOLOA在保持实时性能的同时达到当前最优精度(52.8 / 73.1 mAP on ADG-Det / IIT-Heat),最高速度可达89.77 FPS,轻量版本甚至达846.24 FPS,实现了精度与效率的出色平衡。
原文摘要 · Abstract (English)
Affordance detection aims to jointly address the fundamental "what-where-how" challenge in embodied AI by understanding "what" an object is, "where" the object is located, and "how" it can be used. However, most affordance learning methods focus solely on "how" objects can be used while neglecting the "what" and "where" aspects. Other affordance detection methods treat object detection and affordance learning as two independent tasks, lacking effective interaction and real-time capability. To overcome these limitations, we introduce YOLO Affordance (YOLOA), a real-time affordance detection model that jointly handles these two tasks via a large language model (LLM) adapter. Specifically, YOLOA employs a lightweight detector consisting of object detection and affordance learning branches refined through the LLM Adapter. During training, the LLM Adapter interacts with object and affordance preliminary predictions to refine both branches by generating more accurate class priors, box offsets, and affordance gates. Experiments on our relabeled ADG-Det and IIT-Heat benchmarks demonstrate that YOLOA achieves state-of-the-art accuracy (52.8 / 73.1 mAP on ADG-Det / IIT-Heat) while maintaining real-time performance (up to 89.77 FPS, and up to 846.24 FPS for the lightweight variant). This indicates that YOLOA achieves an excellent trade-off between accuracy and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。