Regression versus classification for neural network based audio source localization

Lauréline Perotin (1) , Alexandre Défossez (2) , Emmanuel Vincent (1) , Romain Serizel (1) , Alexandre Guérin (3)

1 MULTISPEECH - Speech Modeling for Facilitating Oral-Based Communication
2 LIENS - Laboratoire d'informatique de l'école normale supérieure
3 Orange Labs [Cesson-Sévigné]

We compare the performance of regression and classification neural networks for single-source direction-of-arrival estimation. Since the output space is continuous and structured, regression seems more appropriate. However, classification on a discrete spherical grid is widely believed to perform better and is predominantly used in the literature. For regression, we propose two ways to account for the spherical geometry of the output space based either on the angular distance between spherical coordinates or on the mean squared error between Cartesian coordinates. For classification, we propose two alternatives to the classical one-hot encoding framework: we derive a Gibbs distribution from the squared angular distance between grid points and use the corresponding probabilities either as soft targets or as cross-entropy weights that retain a clear probabilis-tic interpretation. We show that regression on Cartesian coordinates is generally more accurate, except when localized interference is present, in which case classification appears to be more robust.

Mots clés: Cost-sensitive classification; training criterion; Direction-of-arrival; soft target; angular loss

Domaines: Traitement du signal et de l'image [eess.SP]; Réseau de neurones [cs.NE]

Lauréline Perotin, Alexandre Défossez, Emmanuel Vincent, Romain Serizel, Alexandre Guérin. Regression versus classification for neural network based audio source localization. WASPAA 2019 - IEEE Workshop on Applications of Signal Processing to Audio and Acoustics, IEEE, Oct 2019, New Paltz, United States. ⟨hal-02125985v2⟩