Adaptive Training of INRs via Pruning and Densification

TL;DR

AIRe is an adaptive training method for implicit neural representations that dynamically prunes neurons and densifies input frequencies, resulting in smaller models with comparable or better reconstruction quality.

Contribution

It introduces a novel adaptive training scheme combining neuron pruning and input frequency densification for improved INR efficiency.

Findings

Reduces model size without sacrificing quality

Improves reconstruction accuracy through adaptive spectrum expansion

Demonstrates effectiveness on images and SDFs

Abstract

Encoding input coordinates with sinusoidal functions into multilayer perceptrons (MLPs) has proven effective for implicit neural representations (INRs) of low-dimensional signals, enabling the modeling of high-frequency details. However, selecting appropriate input frequencies and architectures while managing parameter redundancy remains an open challenge, often addressed through heuristics and heavy hyperparameter optimization schemes. In this paper, we introduce AIRe ($\textbf{A}$daptive $\textbf{I}$mplicit neural $\textbf{Re}$presentation), an adaptive training scheme that refines the INR architecture over the course of optimization. Our method uses a neuron pruning mechanism to avoid redundancy and input frequency densification to improve representation capacity, leading to an improved trade-off between network size and reconstruction quality. For pruning, we first identify less-contributory neurons and apply a targeted weight decay to transfer their information to the remaining neurons, followed by structured pruning. Next, the densification stage adds input frequencies to spectrum regions where the signal underfits, expanding the representational basis. Through experiments on images and SDFs, we show that AIRe reduces model size while preserving, or even improving, reconstruction quality. Code and pretrained models will be released for public use.