Learning Robust Stereo Matching in the Wild with Selective Mixture-of-Experts
About
Recently, learning-based stereo matching networks have advanced significantly. However, they often lack robustness and struggle to achieve impressive cross-domain performance due to domain shifts and imbalanced disparity distributions among diverse datasets. Leveraging Vision Foundation Models (VFMs) can intuitively enhance the model's robustness, but integrating such a model into stereo matching cost-effectively to fully realize their robustness remains a key challenge. To address this, we propose SMoEStereo, a novel framework that adapts VFMs for stereo matching through a tailored, scene-specific fusion of Low-Rank Adaptation (LoRA) and Mixture-of-Experts (MoE) modules. SMoEStereo introduces MoE-LoRA with adaptive ranks and MoE-Adapter with adaptive kernel sizes. The former dynamically selects optimal experts within MoE to adapt varying scenes across domains, while the latter injects inductive bias into frozen VFMs to improve geometric feature extraction. Importantly, to mitigate computational overhead, we further propose a lightweight decision network that selectively activates MoE modules based on input complexity, balancing efficiency with accuracy. Extensive experiments demonstrate that our method exhibits state-of-the-art cross-domain and joint generalization across multiple benchmarks without dataset-specific adaptation. The code is available at \textcolor{red}{https://github.com/cocowy1/SMoE-Stereo}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Stereo Matching | KITTI 2015 | -- | 118 | |
| Stereo Matching | KITTI 2012 | Error Rate (3px, All)4.23 | 108 | |
| Stereo Matching | ETH3D | Threshold Error > 1px (Noc)2.08 | 50 | |
| Stereo Matching | KITTI 2015 (all pixels) | D1 Error (Background)1.5 | 48 | |
| Stereo Matching | ETH3D (non-occluded) | Bad 1.0 Error0.95 | 43 | |
| Stereo Matching | KITTI Noc 2015 | D1 Error (Background)1.4 | 42 | |
| Stereo Matching | Booster Q | EPE2.8 | 33 | |
| Stereo Matching | Middlebury 2021 | Bad Pixel Rate (Thresh > 2.0, All)21.02 | 24 | |
| Stereo Matching | DrivingStereo Zero-shot generalization | Error Rate (Sunny)3.51 | 15 | |
| Stereo Matching | Middlebury non-occluded Half resolution | D27.06 | 14 |