الكشف عن الأجسام البارزة باستخدام RGB-D من منظور التوازن بين السرعة والدقة Lightweight RGB-D Salient Object Detection From a Speed-Accuracy Tradeoff Perspective

شارك:
المجلة: IEEE Transactions on Image Processing، المجلد: 34
DOI: https://doi.org/10.1109/tip.2025.3560488
PMID: https://pubmed.ncbi.nlm.nih.gov/40249695
تاريخ النشر: 2025-01-01
المؤلف: Songsong Duan وآخرون
الموضوع الرئيسي: الانتباه البصري واكتشاف الأهمية

نظرة عامة

في هذه الورقة، يقدم المؤلفون شبكة تبادل السرعة والدقة (SATNet) التي تهدف إلى تعزيز اكتشاف الكائنات البارزة RGB-D الخفيف الوزن (SOD) مع الحفاظ على أداء عالٍ. تعالج SATNet التبادل الشائع بين الدقة والكفاءة في الأساليب الحالية من خلال التركيز على ثلاثة مجالات رئيسية: جودة العمق، دمج الأنماط، وتمثيل الميزات. يتم استخدام نموذج Depth Anything لتوليد خرائط عمق متفوقة، مما يقلل من المشكلات المتعلقة ببيانات العمق الضعيفة في مجموعات البيانات الحالية. بالإضافة إلى ذلك، تم تصميم وحدة الانتباه المفصول (DAM) لتحليل الميزات متعددة الأنماط من خلال فصلها إلى متجهات عرض مزدوجة، مما يسهل استخراج المعلومات بشكل أفضل. تعزز وحدة تمثيل المعلومات المزدوجة (DIRM) مساحة الميزات من خلال نمذجة كل من ميزات القوام والبارزة، باستخدام إطار ثنائي الاتجاه لضبط المعلمات بشكل مثالي.

تظهر النتائج من تجارب واسعة عبر خمسة مجموعات بيانات عامة RGB-D SOD أن SATNet يتفوق على النماذج الثقيلة القائمة على CNN بينما يحافظ على هيكل خفيف الوزن مع 5.2 مليون معلمة فقط ويحقق 415 إطارًا في الثانية (FPS). كما يبرز المؤلفون إمكانية دمج نماذج الرؤية الأساسية على نطاق واسع، مثل نموذج Segment Anything (SAM)، في مجال SOD، مع الاعتراف بالتحديات التي تطرحها الإفراط في التقسيم وغموض القناع. ستركز الأعمال المستقبلية على تحسين دمج هذه النماذج الكبيرة من خلال تقنيات مثل التقطير والتعديل الدقيق، مبتعدين عن استراتيجيات التحفيز غير التعليمية الحالية المرتبطة بـ SAM.

طرق

في هذا القسم، يوضح المؤلفون منهجيتهم لتقييم نموذج SATNet المقترح مقابل 16 طريقة رائدة (SOTA) عبر خمسة مجموعات بيانات عامة RGB-D. يتضمن التقييم مقارنات كمية ونوعية. من الناحية الكمية، يظهر SATNet أداءً متفوقًا من حيث مقياس الخطأ المطلق المتوسط (MAE)، محققًا مكاسب أداء كبيرة بنسبة 20.8%، 23.7%، 10.2%، 40.7%، و28.6% مقارنة بالطريقة الخفيفة السابقة SOTA LSNet عبر مجموعات البيانات NLPR، NJU2K، SIP، STERE، وRGBD135. كما يظهر SATNet أسرع سرعة استدلال بين النماذج الخفيفة، مع مقارنة ملحوظة ضد AirSOD (365 مقابل 415 مللي ثانية). علاوة على ذلك، يتفوق SATNet على الطرق الثقيلة القائمة على CNN مثل HiDANet بينما يحتفظ فقط بـ 4.0% من معلماتها و2.1% من FLOPs، مما يشير إلى توازن ناجح بين الأداء والكفاءة.

نوعيًا، تكشف المقارنات البصرية أن SATNet ينتج خرائط بارزة مع تحديد أفضل وحواف أكثر حدة، مما يتيح تقسيم الكائنات الأمامية بشكل فعال حتى في ظروف صعبة مثل التباين المنخفض والخلفيات المعقدة. كما يبرز المؤلفون فعالية آلية الانتباه المزدوج المقترحة (DAM)، التي تعزز مقاييس الأداء عبر تكوينات مختلفة. بالإضافة إلى ذلك، يتم مقارنة SATNet بأساليب قائمة على المحولات، مما يظهر نتائج تنافسية مع عدد أقل بكثير من المعلمات وFLOPs، مما يثبت كفاءته وفعاليته في استخراج الميزات متعددة الأنماط. بشكل عام، تؤكد النتائج قدرة SATNet على التفوق على الأساليب الحالية مع الحفاظ على هيكل خفيف الوزن.

نقاش

يسلط قسم النقاش في ورقة البحث الضوء على التحديات الحرجة في طرق اكتشاف الكائنات البارزة RGB-D الخفيفة الوزن (SOD)، مع التركيز بشكل خاص على دمج الأنماط غير الكافي وتمثيل الميزات المقيد. يشير المؤلفون إلى أنه بينما تستخدم الأساليب الثقيلة آليات انتباه متنوعة (مثل القناة، المكاني، والانتباه الذاتي) لدمج معلومات RGB والعمق بشكل فعال، تؤدي هذه الآليات إلى دمج غير كافٍ للأنماط في الأطر الخفيفة، مما يؤدي إلى أداء دون المستوى. بالإضافة إلى ذلك، تحد الشبكات الخلفية الخفيفة من تمثيل الميزات، وهو أمر حاسم لأداء الاستدلال، كما يتضح من العلاقة الإيجابية بين غنى مساحة الميزات وأداء النموذج.

لمعالجة هذه القضايا، يقترح المؤلفون شبكة تبادل السرعة والدقة (SATNet)، التي تدمج عدة وحدات مبتكرة. يتم تقديم نموذج Depth Anything لتوليد خرائط عمق عالية الجودة، مما يعزز المعلومات العميقة المستخدمة في SOD. تم تصميم وحدة الانتباه المفصول (DAM) لدمج ميزات RGB والعمق بكفاءة من خلال فصل الميزات الخاصة بالنمط إلى متجهات عرض مزدوجة، مما يحسن التفاعلات عبر الأنماط. علاوة على ذلك، تعزز وحدة تمثيل المعلومات المزدوجة (DIRM) مساحة الميزات المقيدة من خلال استخراج ميزات القوام والبارزة من خلال رؤوس توقع ثنائية الاتجاه، مما يسهل التحسين الثنائي الاتجاه. ثم تدمج وحدة تجميع الميزات المزدوجة (DFAM) هذه الميزات لإنتاج خرائط بارزة أكثر دقة. تظهر النتائج التجريبية عبر خمسة مجموعات بيانات عامة RGB-D SOD أن SATNet يتفوق على الأساليب الخفيفة الحالية، محققًا دقة متفوقة مع عدد أقل بكثير من المعلمات وسرعات استدلال أسرع.

Journal: IEEE Transactions on Image Processing, Volume: 34
DOI: https://doi.org/10.1109/tip.2025.3560488
PMID: https://pubmed.ncbi.nlm.nih.gov/40249695
Publication Date: 2025-01-01
Author(s): Songsong Duan et al.
Primary Topic: Visual Attention and Saliency Detection

Overview

In this paper, the authors introduce the Speed-Accuracy Tradeoff Network (SATNet) aimed at enhancing lightweight RGB-D salient object detection (SOD) while maintaining high performance. SATNet addresses the common trade-off between accuracy and efficiency in existing methods by focusing on three key areas: depth quality, modality fusion, and feature representation. The Depth Anything Model is employed to generate superior depth maps, mitigating issues related to poor depth data in current datasets. Additionally, the Decoupled Attention Module (DAM) is designed to analyze multi-modal features by separating them into dual-view vectors, facilitating better information extraction. The Dual Information Representation Module (DIRM) further enriches the feature space by modeling both texture and saliency features, utilizing a bi-directional framework for optimal parameter tuning.

The results from extensive experiments across five public RGB-D SOD datasets demonstrate that SATNet outperforms state-of-the-art CNN-based heavyweight models while maintaining a lightweight architecture with only 5.2 million parameters and achieving 415 frames per second (FPS). The authors also highlight the potential of integrating large-scale vision foundation models, such as the Segment Anything Model (SAM), into the SOD domain, while acknowledging the challenges posed by over-segmentation and mask ambiguity. Future work will focus on refining the integration of these large models through techniques like distillation and fine-tuning, moving away from the current non-learning prompt strategies associated with SAM.

Methods

In this section, the authors detail their methodology for evaluating the proposed SATNet model against 16 state-of-the-art (SOTA) methods across five public RGB-D datasets. The evaluation includes both quantitative and qualitative comparisons. Quantitatively, SATNet demonstrates superior performance in terms of the Mean Absolute Error (MAE) metric, achieving significant performance gains of 20.8%, 23.7%, 10.2%, 40.7%, and 28.6% over the previous lightweight SOTA method LSNet across the datasets NLPR, NJU2K, SIP, STERE, and RGBD135. SATNet also exhibits the fastest inference speed among lightweight models, with a notable comparison against AirSOD (365 vs. 415 ms). Furthermore, SATNet outperforms heavyweight CNN-based methods like HiDANet while maintaining only 4.0% of their parameters and 2.1% of their FLOPs, indicating a successful balance between performance and efficiency.

Qualitatively, the visual comparisons reveal that SATNet produces saliency maps with better localization and sharper edges, effectively segmenting foreground objects even in challenging conditions such as low contrast and complex backgrounds. The authors also highlight the effectiveness of their proposed Dual Attention Mechanism (DAM), which enhances performance metrics across various configurations. Additionally, SATNet is compared with transformer-based methods, showing competitive results with significantly fewer parameters and FLOPs, further validating its efficiency and effectiveness in multi-modality feature extraction. Overall, the results underscore SATNet’s capability to outperform existing methods while maintaining a lightweight architecture.

Discussion

The discussion section of the research paper highlights critical challenges in lightweight RGB-D salient object detection (SOD) methods, particularly focusing on inadequate modality fusion and constrained feature representation. The authors note that while heavyweight methods utilize various attention mechanisms (e.g., channel, spatial, and self-attention) to effectively fuse RGB and depth information, these mechanisms lead to insufficient modality fusion in lightweight frameworks, resulting in subpar performance. Additionally, lightweight backbone networks limit feature representation, which is crucial for inference performance, as evidenced by the positive correlation between feature space richness and model performance.

To address these issues, the authors propose the Speed-Accuracy Tradeoff Network (SATNet), which integrates several innovative modules. The Depth Anything Model is introduced to generate high-quality depth maps, enhancing the depth information utilized in SOD. The Decoupled Attention Module (DAM) is designed to efficiently fuse RGB and depth features by decoupling modality-specific features into dual-view vectors, thereby improving cross-modal interactions. Furthermore, the Dual Information Representation Module (DIRM) enriches the constrained feature space by extracting texture and saliency features through two-way prediction heads, facilitating bidirectional optimization. The Dual Feature Aggregation Module (DFAM) then integrates these features to produce more accurate saliency maps. Experimental results across five public RGB-D SOD datasets demonstrate that SATNet outperforms existing lightweight approaches, achieving superior accuracy with significantly fewer parameters and faster inference speeds.

شارك: