Fast SceneScript: Accurate and Efficient Structured Language Model via Multi-Token Prediction
By: Ruihong Yin , Xuepeng Shi , Oleksandr Bailo and more
Recent perception-generalist approaches based on language models have achieved state-of-the-art results across diverse tasks, including 3D scene layout estimation, via unified architecture and interface. However, these approaches rely on autoregressive next-token prediction, which is inherently slow. In this work, we introduce Fast SceneScript, a novel structured language model for accurate and efficient 3D scene layout estimation. Our method employs multi-token prediction (MTP) to reduce the number of autoregressive iterations and significantly accelerate inference. While MTP improves speed, unreliable token predictions can significantly reduce accuracy. To filter out unreliable tokens, we adapt self-speculative decoding (SSD) for structured language models and introduce confidence-guided decoding (CGD) with an improved scoring mechanism for token reliability. Furthermore, we design a parameter-efficient mechanism that reduces the parameter overhead of MTP. Extensive experiments on the ASE and Structured3D benchmarks demonstrate that Fast SceneScript can generate up to 9 tokens per decoder inference step without compromising accuracy, while adding only $\sim7.5\%$ additional parameters.
Similar Papers
FastMTP: Accelerating LLM Inference with Enhanced Multi-Token Prediction
Machine Learning (CS)
Makes AI write much faster without mistakes.
Script: Graph-Structured and Query-Conditioned Semantic Token Pruning for Multimodal Large Language Models
CV and Pattern Recognition
Makes AI see pictures and videos faster.
Scenes as Tokens: Multi-Scale Normal Distributions Transform Tokenizer for General 3D Vision-Language Understanding
CV and Pattern Recognition
Helps computers understand 3D worlds like we do.