εar-VAE2

Fourier is Frontier: Frequency-Aware Autoencoding
for High-Fidelity Music Reconstruction
Anonymous authors

A spec-domain autoencoder compressing 48 kHz stereo audio into a 25 Hz sequence of continuous latents — a 1920× temporal downsampling — through two frequency-aware components: Spec-SnakeBeta and a Duplex-Aware Refiner.

01

Which factors make the LDM-base generation struggling?

Motivation

Latent-diffusion music generators operate on the latent space of a VAE — whatever spectral detail, phase structure, or stereo image the decoder cannot reconstruct is unavailable to any downstream generator. At the aggressive temporal compression these systems demand, decoders show three recurring failure modes. We ask: at a fixed compression rate, which modeling choices best preserve what matters?

01

High-frequency detail lost

A channel-wise activation reuses one response across all spectral bins and cannot adapt directly to physical frequency.

→ Spec-SnakeBeta · 03
02

Phase smeared across frames

Phase error grows monotonically with frequency and saturates near random (π/2) above 4 kHz.

→ Complex STFT architecture · 02
03

Stereo image collapses

A uniform corrector spends capacity in the wrong place; localization cues differ across the spectrum.

→ Duplex-Aware Refiner · 04
02

Why does the architecture work?

Spectral-domain VAE-GAN

Stereo audio is transformed by STFT into a complex spectrogram; a 2D-convolutional Spec Encoder with delayed stereo fusion compresses it to a 25 Hz continuous latent, and a mirrored Spec Decoder with early splitting reconstructs a coarse spectrogram. The Duplex-Aware Refiner then applies band-specific complex corrections and restores the Nyquist bin. A paired iSTFT then removes the explicit analysis margin and returns exactly the original number of waveform samples. Click a stage to inspect it.

PIPELINEεar-VAE2 at a glance

Waveform → STFT → complex spectrogram → encoder → 25 Hz latent → decoder → duplex-aware refinement → iSTFT.

03

How should activations adapt to frequency?

Frequency-axis periodic activation

BigVGAN’s SnakeBeta activation, x + β⁻¹ sin²(αx), learns its periodicity per channel — in the spectral domain that mixes all frequencies onto one curve. Because each STFT bin corresponds to a fixed physical frequency, Spec-SnakeBeta learns one (αf, βf) pair per frequency bin and shares it across feature channels. A frequency-proportional initialization in log-parameter space starts near identity at low bins and becomes progressively more oscillatory toward high frequencies. Drag the frequency slider to inspect it.

From Snake to Spec-SnakeBeta.  Standard activations such as ReLU and GELU do not provide an explicit periodic basis. On bounded intervals a neural network can still approximate periodic functions; the Snake family instead supplies a periodic residual directly through three design stages:

I

Snake

y = x + α−1 sin²(α · x)

A single parameter α controls both oscillation frequency and inverse amplitude (1/α) — the two degrees of freedom are optimization-coupled.

II

SnakeBeta

y = x + β−1 sin²(α · x)

Decouples frequency (α) from amplitude (β) with independent gradient paths. Per-channel — all frequency bins share one (αβ) pair.

III

Spec-SnakeBeta

yc,f = xc,f + βf−1 sin²(αf · xc,f)

One log-space response per physical frequency, shared across feature channels — near-identity at low bins and more oscillatory at high.

yc,f = xc,f + βf−1 sin²(αf · xc,f),  αf(0) ≈ freqf / f̄
frequency 6.0 kHz
parameterization
log-F αf initialization
Top: activation curves. Faint curves show other frequencies; the dashed line is the identity. Bottom: the frequency-proportional initialization αf = freqf/f̄ over the 480 bins, stored in log-parameter space (f̄ ≈ 11.98 kHz). After training, α adapts most strongly in 0–5 kHz (shaded). “Channel” here means a hidden feature channel, not the left/right audio channels. Spec-SnakeBeta(CF) starts from the same frequency curve as Spec-SnakeBeta(log-F) but optimizes a separate copy for every (channel, frequency) pair; see the ablation below.
Ablation — seven activation configurations (controlled mid-scale, 42.6M generator)

LOG-FSpec-SnakeBeta(log-F)

Frequency-indexed and shared across feature channels. The initial α response is proportional to normalized physical frequency, while β starts at 1; both positive parameters are optimized in log space.

Channel sharing is an inductive bias, not just a compression trick. Spec-SnakeBeta(log-F) learns one response per frequency and shares it across feature channels. It leads the fully independent Spec-SnakeBeta(CF) alternative on all six three-seed mean metrics while using far fewer activation parameters, and achieves the best overall mean on four of six metrics.
04

Where should the refiner spend its capacity?

Psychoacoustically placed corrections

Human localization relies on interaural timing (phase) at low frequencies and interaural level (magnitude) at high frequencies — the classical duplex theory (Rayleigh, 1907). Per-frequency error analysis of the trained decoder shows the same asymmetry: magnitude error peaks in the 1–4 kHz mid band while phase error saturates above 4 kHz. The refiner therefore constrains its residual outputs to the cue emphasized in each band. Its output layers are zero-initialized, so refinement starts as an identity mapping that preserves the frozen decoder output. Hover the bands.

Band placement on the physical frequency axis

DUPLEXThree bands, three correction types

Phase-only below 1.5 kHz · joint magnitude & phase in 1.5–4 kHz · magnitude-only above 4 kHz.

Why the bands sit where they do — per-frequency error profiles (schematic)
Schematic redraw of the per-frequency analysis on the trained Stage-2 decoder: log-magnitude error peaks broadly in the 1–4 kHz mid band, while phase error grows monotonically and approaches π/2 above 4 kHz. Dashed lines mark the 1.5 / 4 kHz band boundaries.
Ablation — banded vs unconstrained refinement
Controlled refiner-only setting from the same frozen decoder. The banded head predicts ≈45% fewer residual dimensions, improves both spectral distances and three of four targeted spatial-cue errors, and receives a higher mean paired score from professional engineers; HF IPD remains within 0.6%. IPD/ILD are mean absolute deviations below 1.5 kHz (LF) and above 4 kHz (HF); the paired 1–10 rating is mapped to [0,1].
05

When does the metric work?

Reconstruction · generation

Reconstruction on the 546-track Song Describer Dataset against four recent open-source audio VAEs, each at its native operating condition. The chart reports point estimates; cross-system numbers are competitive rather than strictly matched; the base-vs-full comparison within εar-VAE2 is controlled.

Main reconstruction results

48 KHZ · 25 HZεar-VAE2 (full)

Runs at 48 kHz with a 25.0 Hz continuous latent. This is the complete decoder with the Duplex-Aware Refiner; the base εar-VAE2 result uses the same operating condition.

Downstream generation — SongBench
εar-VAE2 (full) εar-VAE2 (w/o Refiner) Levo 2 VAE
Axis range 5–8 for readability. 100 EN + 100 ZH generated songs, matched conditioning and CFG; the renderer is retrained per latent space.
Downstream generation — SongEval
εar-VAE2 (full) εar-VAE2 (w/o Refiner) Levo 2 VAE
Truncated axis 3.5–4.6. The refiner improves every axis further over the base decoder — refinement benefits generated outputs, not only reconstruction.
06

What does the latent organize along time?

Temporal-frequency probe

Musical content varies at markedly different rates in time: percussion, onsets, and rhythmic groove change fast, while sustained harmony and timbre change slowly. A latent that mirrors acoustic structure might therefore route fast and slow temporal variation to different content types. To test this, we encode a clip to its latent z ∈ ℝC×T, take the real FFT along the temporal axis, and split the temporal-frequency bins at the midpoint — zeroing the upper half leaves a slow-varying latent zlow, zeroing the lower half a fast-varying latent zhigh — then inverse-FFT and decode each half separately. Pick a model below to compare the decoded spectral profiles.

ρ = centroid(zlow → audio) / centroid(zhigh → audio)
The split, decoded per model

PROBE

Schematic animation of the probe procedure; the centroid and ρ values shown are measured 3-track averages. Click a stage of the diagram for an explanation.
The two acoustically supervised models are inverted (ρ > 1) — εar-VAE2 most strongly (ρ = 2.32), εar-VAE next (ρ = 1.82) — while Levo 2, SAME-L, and SA-Open are not (ρ < 1). For εar-VAE2, the fast-varying half has a lower spectral centroid and is more bass-concentrated than the slow-varying half. This is a correlational diagnostic, not a semantic decomposition, causal mechanism, or quality claim: it rests on 3 tracks, spectral centroid measures brightness rather than rhythm or onsets, model configurations differ, and decoding a band-limited latent is outside the decoder's training distribution.
Quantitative summary — inversion ratio ρ
Spectral centroid of the separately decoded low- / high-temporal-frequency latent halves, averaged over 3 tracks. ρ = centroidlow / centroidhigh; ρ > 1 ("inverted") means the fast-varying latent carries lower-pitched, bass-concentrated audio.
07

Hear the difference

Listen & compare

Switch between versions while playing — playback position is preserved for direct A/B comparison. Gen Demos are full-length songs generated on εar-VAE2 latents; reconstruction clips are 30 s 48 kHz stereo excerpts from the Song Describer Dataset; downstream generation clips are 45 s matched-window pairs. Every clip is loudness-normalized to −14 LUFS (EBU R128 linear gain, true peak ≤ −1.5 dBTP) so comparisons are level-fair.

Gen Demos

full songs generated on εar-VAE2 latents
↑ §05 results

Reconstruction

encode → decode fidelity · 6 systems
↑ §05 results

Downstream generation

Levo 2 VAE vs εar-VAE2 · same renderer architecture, retrained per latent space
↑ §05 results

Latent Temporal-Frequency Probe

Split the latent into slow- and fast-varying halves and decode each separately. Compare the measured spectral-centroid inversion; the probe is exploratory and does not establish a rhythm/content decomposition.
↑ §06 probe discussion

Refiner Ablation

Ground truth vs without refiner vs with refiner — selected reconstruction examples
↑ §04 refiner

Banded Refiner Ablation

Banded (ours) vs unconstrained refiner · same frozen decoder
↑ §04 refiner

The stereo polar field visualization in each player is adapted from the open-source EAR-Audio-Preview project (polar sample mode) by Epsilon Acoustic Revolution Lab.