Skip to main navigation Skip to search Skip to main content

Rethinking the refinement stage of 3D object detection: A multi-task learning perspective with Mixture-of-Experts

  • Bingqian Wu
  • , Pei An
  • , Siwen Quan
  • , Qiao Wu
  • , Linjie Li
  • , Chu'ai Zhang
  • , Jiaqi Yang
  • Northwestern Polytechnical University Xian
  • Huazhong University of Science and Technology
  • Chang'an University

Research output: Contribution to journalArticlepeer-review

Abstract

Two-stage LiDAR-based 3D object detectors have achieved state-of-the-art accuracy, yet their performance is often limited by the refinement stage. In this work, we revisit 3D object refinement from a multi-task learning perspective and identify two independent sources of negative transfer: an inter-attribute conflict, where heterogeneous regression objectives, such as center, size, and orientation, interfere during joint optimization, and an inter-sample conflict, where proposals with varying point densities lead to gradient imbalance. To address these issues, we introduce two specialized Mixture-of-Experts architectures. The Attribute-MoE decouples regression objectives into dedicated expert branches to alleviate feature conflicts, while the Sparsity-MoE employs density-aware experts to adaptively refine proposals according to point sparsity. Integrated into strong two-stage baselines, our modules consistently improve performance on the KITTI dataset and the Waymo Open Dataset. Beyond empirical gains, our analysis reveals that Attribute-MoE and Sparsity-MoE solve largely independent problems, offering a practical “toolbox” for mitigating negative transfer in 3D object refinement and advancing adaptive, task-aware detector design. Code will be released at https://github.com/12e21/RefineMoE.

Original languageEnglish
Article number104841
JournalJournal of Visual Communication and Image Representation
Volume118
DOIs
StatePublished - Jun 2026

Keywords

  • 3D object detection
  • LiDAR
  • Mixture-of-Experts
  • Multi-task learning
  • Negative transfer
  • Point cloud
  • Refinement

Fingerprint

Dive into the research topics of 'Rethinking the refinement stage of 3D object detection: A multi-task learning perspective with Mixture-of-Experts'. Together they form a unique fingerprint.

Cite this