MoE3D: A Mixture-of-Experts Module for 3D Reconstruction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Zichen, Cao, Ang, Wang, Liam J., Park, Jeong Joon
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910000639639552
author Wang, Zichen
Cao, Ang
Wang, Liam J.
Park, Jeong Joon
author_facet Wang, Zichen
Cao, Ang
Wang, Liam J.
Park, Jeong Joon
contents We propose a simple yet effective approach to enhance the performance of feed-forward 3D reconstruction models. Existing methods often struggle near depth discontinuities, where standard regression losses encourage spatial averaging and thus blur sharp boundaries. To address this issue, we introduce a mixture-of-experts formulation that handles uncertainty at depth boundaries by combining multiple smooth depth predictions. A softmax weighting head dynamically selects among these hypotheses on a per-pixel basis. By integrating our mixture model into a pre-trained state-of-the-art 3D model, we achieve a substantial reduction of boundary artifacts and gains in overall reconstruction accuracy. Notably, our approach is highly compute efficient, delivering generalizable improvements even when fine-tuned on a small subset of training data while incurring only negligible additional inference computation, suggesting a promising direction for lightweight and accurate 3D reconstruction.
format Preprint
id arxiv_https___arxiv_org_abs_2601_05208
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle MoE3D: A Mixture-of-Experts Module for 3D Reconstruction
Wang, Zichen
Cao, Ang
Wang, Liam J.
Park, Jeong Joon
Computer Vision and Pattern Recognition
We propose a simple yet effective approach to enhance the performance of feed-forward 3D reconstruction models. Existing methods often struggle near depth discontinuities, where standard regression losses encourage spatial averaging and thus blur sharp boundaries. To address this issue, we introduce a mixture-of-experts formulation that handles uncertainty at depth boundaries by combining multiple smooth depth predictions. A softmax weighting head dynamically selects among these hypotheses on a per-pixel basis. By integrating our mixture model into a pre-trained state-of-the-art 3D model, we achieve a substantial reduction of boundary artifacts and gains in overall reconstruction accuracy. Notably, our approach is highly compute efficient, delivering generalizable improvements even when fine-tuned on a small subset of training data while incurring only negligible additional inference computation, suggesting a promising direction for lightweight and accurate 3D reconstruction.
title MoE3D: A Mixture-of-Experts Module for 3D Reconstruction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.05208