Stereo2Spatial: Convert Stereo Music Tracks to Spatialized Binaural Mixes [P]

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, short

Summary

The Stereo2Spatial model, developed over approximately six months, converts standard stereo music tracks into spatialized binaural mixes. Initially, a flow-matching diffusion model was built in the latent space of an EAR-VAE, encoding stereo input into one latent and 7.1.4 output channels. This latent approach, inspired by ImmersiveFlow, included memory tokens for stable long-context generations but faced quality bottlenecks due to VAE distribution mismatch. Pivoting to a raw waveform model resolved quality issues but introduced training instability, with loss shooting up after 60K-80K steps. This was overcome by implementing "amplitude lifting" from the WavFlow paper, scaling audio tracks to an RMS of 0.33 and multiplying by 3, using a clip of 4.0 for a model space of -12 to 12. The waveform model was trained on 7,669 tracks for about 20 days on 2x A6000 GPUs, with optional mix-style conditioning. A Windows desktop application is also available for inference.

Key takeaway

For AI Engineers or audio developers exploring spatial audio generation, the Stereo2Spatial model offers a robust open-source solution for converting stereo to binaural mixes. You should consider integrating this Apache 2.0 licensed model, especially noting the effectiveness of amplitude lifting for training stability and the availability of a Windows inference app. Evaluate its direct binaural output and explore mix-style conditioning for your projects.

Key insights

Converting stereo to spatial audio via flow-matching diffusion is feasible, with raw waveform modeling and amplitude lifting crucial for quality and stability.

Principles

Method

The waveform model trains in two stages on 7,669 tracks, using amplitude lifting (RMS 0.33, clip 4.0) to stabilize flow-matching diffusion.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.