paper-with-me

홈 › Papers

M2H: Multi-Task Learning with Efficient Window-Based Cross-Task Attention for Monocular Spatial Perception

2025-10-20 · U. V. B. L Udugama, George Vosselman, Francesco Nex arxiv

Deploying real-time spatial perception on edge devices requires efficient multi-task models that leverage complementary task information while minimizing computational overhead. This paper introduces Multi-Mono-Hydra (M2H), a novel multi-task learning framework designed for semantic segmentation and depth, edge, and surface normal estimation from a single monocular image. Unlike conventional approaches that rely on independent single-task models or shared encoder-decoder architectures, M2H introduces a Window-Based Cross-Task Attention Module that enables structured feature exchange while preserving task-specific details, improving prediction consistency across tasks. Built on a lightweight ViT-based DINOv2 backbone, M2H is optimized for real-time deployment and serves as the foundation for monocular spatial perception systems supporting 3D scene graph construction in dynamic environments. Comprehensive evaluations show that M2H outperforms state-of-the-art multi-task models on NYUDv2, surpasses single-task depth and semantic baselines on Hypersim, and achieves superior performance on the Cityscapes dataset, all while maintaining computational efficiency on laptop hardware. Beyond benchmarks, M2H is validated on real-world data, demonstrating its practicality in spatial perception tasks.

📄 PDF Abstract BibTeX arXiv:2510.17363

Code (0)

등록된 구현이 없습니다.

Tasks

Computational EfficiencySemantic SegmentationMulti-Task Learning

Similar Papers 제목 키워드 기반

MSWA: Refining Local Attention with Multi-ScaleWindow Attention

2025-01-02 · Yixing Xu, Shivank Nag, Dong Li, Lu Tian 외

Transformer-based LLMs have achieved exceptional performance across a wide range of NLP tasks. However, the standard self-attention mechanism suffers from quadratic time complexity and linearly increased cache size. Slid…

Common Sense ReasoningLanguage ModelingLanguage Modelling

AgileIR: Memory-Efficient Group Shifted Windows Attention for Agile Image Restoration

2024-09-10 · Hongyi Cai, Mohammad Mahdinur Rahman, Mohammad Shahid Akhtar, Jie Li 외

Image Transformers show a magnificent success in Image Restoration tasks. Nevertheless, most of transformer-based models are strictly bounded by exorbitant memory occupancy. Our goal is to reduce the memory consumption o…

Image RestorationQuantization

MixFormer: Mixing Features across Windows and Dimensions

2022-04-06 · CVPR 2022 1 · Qiang Chen, Qiman Wu, Jian Wang, Qinghao Hu 외

While local-window self-attention performs notably in vision tasks, it suffers from limited receptive field and weak modeling capability issues. This is mainly because it performs self-attention within non-overlapped win…

image-classificationImage Classification

Differentiable Window for Dynamic Local Attention

2020-06-24 · ACL 2020 6 · Thanh-Tung Nguyen, Xuan-Phi Nguyen, Shafiq Joty, Xiao-Li Li

We propose Differentiable Window, a new neural module and general purpose component for dynamic window selection. While universally applicable, we demonstrate a compelling use case of utilizing Differentiable Window to i…

Language ModelingLanguage ModellingMachine TranslationSentiment Analysis+1

Masked Autoencoders with Multi-Window Local-Global Attention Are Better Audio Learners

2023-06-01 · Sarthak Yadav, Sergios Theodoridis, Lars Kai Hansen, Zheng-Hua Tan

In this work, we propose a Multi-Window Masked Autoencoder (MW-MAE) fitted with a novel Multi-Window Multi-Head Attention (MW-MHA) module that facilitates the modelling of local-global interactions in every decoder trans…

Decoder