Paper Reading AI Learner

Efficient Prediction of Dense Visual Embeddings via Distillation and RGB-D Transformers

2026-01-01 14:29:31
S\"ohnke Benedikt Fischedick, Daniel Seichter, Benedict Stephan, Robin Schmidt, Horst-Michael Gross

Abstract

In domestic environments, robots require a comprehensive understanding of their surroundings to interact effectively and intuitively with untrained humans. In this paper, we propose DVEFormer - an efficient RGB-D Transformer-based approach that predicts dense text-aligned visual embeddings (DVE) via knowledge distillation. Instead of directly performing classical semantic segmentation with fixed predefined classes, our method uses teacher embeddings from Alpha-CLIP to guide our efficient student model DVEFormer in learning fine-grained pixel-wise embeddings. While this approach still enables classical semantic segmentation, e.g., via linear probing, it further enables flexible text-based querying and other applications, such as creating comprehensive 3D maps. Evaluations on common indoor datasets demonstrate that our approach achieves competitive performance while meeting real-time requirements, operating at 26.3 FPS for the full model and 77.0 FPS for a smaller variant on an NVIDIA Jetson AGX Orin. Additionally, we show qualitative results that highlight the effectiveness and possible use cases in real-world applications. Overall, our method serves as a drop-in replacement for traditional segmentation approaches while enabling flexible natural-language querying and seamless integration into 3D mapping pipelines for mobile robotics.

Abstract (translated)

在家庭环境中,机器人需要全面了解周围环境,以便与未经培训的人类用户有效且直观地互动。本文提出了DVEFormer——一种基于RGB-D的Transformer方法,通过知识蒸馏预测密集的文字对齐视觉嵌入(DVE)。不同于直接执行具有固定预定义类别划分的传统语义分割任务,我们的方法使用Alpha-CLIP教师模型生成的嵌入来指导学生模型DVEFormer学习精细粒度的像素级嵌入。尽管这种方法仍然支持传统的语义分割任务(例如通过线性探针进行),它还进一步支持基于文本查询和创建全面的3D地图等其他应用。 在常见的室内数据集上的评估表明,我们的方法实现了竞争性的性能并满足了实时需求,在NVIDIA Jetson AGX Orin平台上运行完整模型时帧率为26.3 FPS,较小变体版本为77.0 FPS。此外,我们展示了定性结果以突出该方法的有效性和可能的应用场景。 总体而言,我们的方法可以作为传统分割方法的直接替代品,并且支持灵活的自然语言查询和无缝集成到移动机器人的3D地图构建管道中。

URL

https://arxiv.org/abs/2601.00359

PDF

https://arxiv.org/pdf/2601.00359.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot