In plain words: A user clicks a wrong spot in a 3D room layout, and the model fills in the fix like filling a missing word in a sentence. Trained on both whole-room prediction and these fixes, it kept its accuracy while getting better at local corrections.
Abstract
We present a novel human-in-the-loop approach to estimate 3D scene layout that uses human feedback from an egocentric standpoint. We study this approach through introduction of a novel local correction task, where users identify local errors and prompt a model to automatically correct them. Building on SceneScript, a state-of-the-art framework for 3D scene layout estimation that leverages structured language, we propose a solution that structures this problem as "infilling", a task studied in natural language processing. We train a multi-task version of SceneScript that maintains performance on global predictions while significantly improving its local correction ability. We integrate this into a human-in-the-loop system, enabling a user to iteratively refine scene layout estimates via a low-friction "one-click fix'' workflow. Our system enables the final refined layout to diverge from the training distribution, allowing for more accurate modelling of complex layouts.
Christopher Xie, Armen Avetisyan, Henry Howard-Jenkins, Yawar Siddiqui, Julian Straub, Richard Newcombe, Vasileios Balntas, Jakob Engel
arXiv:2503.11806 · cs.CV · submitted Mar 14, 2025 · updated Jul 30, 2025
abstract · pdf · html · Project page: https://www.projectaria.com/scenescript/