AIR: Amortized Image Reconstruction Framework for Self-Supervised Feed-Forward 2D Gaussian Splatting
Not on the map. Why?
Abstract (arXiv)
2D Gaussian splatting provides an efficient explicit representation for image reconstruction, but existing methods still require costly per-image iterative optimization or rely on handcrafted priors for primitive allocation. We present AIR, a self-supervised feed-forward framework that amortizes iterative Gaussian fitting into a single network pass, eliminating per-image test-time optimization. AIR adopts a stage-wise residual architecture that progressively predicts additional Gaussian primitives from reconstruction residuals, together with an explicit Stage Control mechanism that activates new primitives only in under-reconstructed regions. A Predict--Optimize--Distill training strategy stabilizes multi-stage prediction by distilling short-horizon optimized Gaussian increments back into the predictor. The stabilized predictor is then jointly finetuned across stages and equipped with an image-adaptive quantizer for compact Gaussian storage. Experiments on Kodak and DIV2K show that AIR achieves better reconstruction quality than representative Gaussian-based baselines while reducing encoding time to 160--300,ms. Code: https://github.com/whoiszzj/AIR.git
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 39 of its 51 references that are in the map. First posted on arXiv 20 May 2026.
Builds on
39 of 51 references.
References · 39
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
- 2022Instant Neural Graphics Primitives with a Multiresolution Hash Encodingcited by 3,723
- 2020Implicit Neural Representations with Periodic Activation Functionscited by 2,238
- 2020Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domainscited by 2,200
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,079
- 2021Mip-NeRF 360: Unbounded Anti-Aliased Neural Radiance Fieldscited by 2,032
- 2025DINOv3cited by 1,414
- 2020Learning Continuous Image Representation with Local Implicit Image Functioncited by 552
- 2024MoGe: Unlocking Accurate Monocular Geometry Estimation for Open-Domain Images with Optimal Training Supervisioncited by 299
- 2020Deep Learning Techniques for Inverse Problems in Imagingcited by 291
27 more
- 2025MoGe-2: Accurate Monocular Geometry with Metric Scale and Sharp Detailscited by 261
- 2017Learned Primal-dual Reconstructioncited by 251
- 2021COIN: COmpression with Implicit Neural representationscited by 233
- 2023WIRE: Wavelet Implicit Neural Representationscited by 220
- 2021Implicit Neural Representations for Image Compressioncited by 162
- 2021BACON: Band-limited Coordinate Networks for Multiscale Scene Representationcited by 143
- 2016Spatially Adaptive Computation Time for Residual Networkscited by 118
- 2024GaussianImage: 1000 FPS Image Representation and Compression by 2D Gaussian Splattingcited by 99
- 2018Iterative Amortized Inferencecited by 94
- 2022Tutorial on amortized optimizationcited by 83
- 2022COIN++: Neural Compression Across Modalitiescited by 82
- 2019Hypernetwork functional image representationcited by 65
- 2021Variable-Rate Deep Image Compression through Spatially-Adaptive Feature Transformcited by 58
- 2021Seeing Implicit Neural Representations as Fourier Seriescited by 50
- 2021Meta-Learning Sparse Implicit Neural Representationscited by 48
- 2024Where Do We Stand with Implicit Neural Representations? A Technical and Performance Surveycited by 42
- 2024Image-GS: Content-Adaptive Image Representation via 2D Gaussianscited by 35
- 2017Recurrent Inference Machines for Solving Inverse Problemscited by 33
- 2025Large Images are Gaussians: High-Quality Large Image Representation with Levels of 2D Gaussian Splattingcited by 28
- 2022Meta-Learning Sparse Compression Networkscited by 22
- 2025Instant GaussianImage: A Generalizable and Self-Adaptive Image Representation via 2D Gaussian Splattingcited by 18
- 2025ReSplat: Learning Recurrent Gaussian Splattingcited by 15
- 2018Spatially adaptive image compression using a tiled deep networkcited by 10
- 2025Predict-Optimize-Distill: A Self-Improving Cycle for 4D Object Understandingcited by 8
- 2025GaussianImage++: Boosted Image Representation and Compression with 2D Gaussian Splattingcited by 5
- 2025Structure-Guided Allocation of 2D Gaussians for Image Representation and Compressioncited by 5
- 2026GIFSplat: Generative Prior-Guided Iterative Feed-Forward 3D Gaussian Splatting from Sparse Viewscited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Momentum-GS: Momentum Gaussian Self-Distillation for High-Quality Large Scene Reconstruction
- 2024AugGS: Self-augmented Gaussians with Structural Masks for Sparse-view 3D Reconstruction
- 2026GenSplatCodec: Feed-Forward Gaussian Splatting Compression via One-Step Diffusion
- 2026ReCoSplat: Online Feed-Forward Gaussian Splatting via Render-and-Compare
- 2026FeCoSplat: Feedback-Guided Compression for Feed-Forward 3D Gaussian Splatting
- 2026SGI: Structured 2D Gaussians for Efficient and Compact Large Image Representation
- 2023Compressed 3D Gaussian Splatting for Accelerated Novel View Synthesis
- 2025Optimized Minimal 4D Gaussian Splatting