LSRM: High-Fidelity Object-Centric Reconstruction via Scaled Context Windows

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, extended

Summary

The Large Sparse Reconstruction Model (LSRM) significantly enhances feed-forward 3D object reconstruction and inverse rendering by scaling transformer context windows. It adapts native sparse attention (NSA) to process 20x more object tokens and >2x more image tokens than previous methods. LSRM utilizes an efficient coarse-to-fine pipeline, a 3D-aware spatial routing mechanism that employs explicit geometric distances, and a custom block-aware sequence parallelism strategy for balanced GPU workloads. Evaluations on novel-view synthesis benchmarks demonstrate substantial gains, including >2.4 dB higher PSNR and >40% lower LPIPS on the GSO dataset. For inverse rendering, LSRM achieves an LPIPS that matches state-of-the-art dense-view optimization methods on StanfordORB.

Key takeaway

For Machine Learning Engineers developing high-fidelity 3D reconstruction or inverse rendering models, consider adopting LSRM's approach. Expanding transformer context windows with sparse attention and a coarse-to-fine pipeline significantly boosts texture and geometry detail. Implement 3D-aware block routing and block-aware sequence parallelism to manage computational demands and achieve state-of-the-art performance, even with fewer input views than typically required.

Key insights

Scaling transformer context windows markedly improves feed-forward 3D reconstruction and inverse rendering fidelity.

Principles

Method

A two-stage coarse-to-fine pipeline initializes with dense reconstruction, then refines with sparse reconstruction predicting high-resolution residuals using NSA, 3D-aware routing, and block-aware sequence parallelism.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.