Aura: Privacy-preserving Augmentation to Improve Test Set Diversity in Noise Suppression Applications
Noise suppression models running in production environments are commonly trained on publicly available datasets. However, this approach leads to regressions in production environments due to the lack of training/testing on representative customer data. Moreover, due to privacy reasons, developers cannot listen to customer content. This "ears-off" situation motivates augmenting existing datasets in a privacy-preserving manner. In this paper, we present Aura, a solution to make existing noise suppression test sets more challenging and diverse while limiting the sampling budget. Aura is "ears-off" because it relies on a feature extractor and a metric of speech quality, DNSMOS P.835, both pre-trained on data obtained from public sources. As an application of Aura, we augment a current benchmark test set in noise suppression by sampling audio files from a new batch of data of 20K clean speech clips from Librivox mixed with noise clips obtained from AudioSet. Aura makes the existing benchmark test set harder by 100% in DNSMOS P.835, a 26% improvement in Spearman's rank correlation coefficient (SRCC) compared to random sampling and, identifies 73% out-of-distribution samples to augment the test set.
The current iteration showcases Aura for the Deep Noise Suppression scenario. However you can utilize for any scenario where you want to create a test set with more challenging and diverse samples in a privacy preserving manner. All you need is a feature extractor, an ontology, a mechanism to cluster, and an objective metric.
Create a conda environment from the requirement.yml file
conda env create --file requirement.yml
You also need git lfs for accessing the audio data which can be installed using the following commands:
git lfs install
For more details, please look at Git Large File Storage.
Data to be uploaded
data_csv: Location of csv containing paths to data files data/data.csv
n: Number of samples you want (default: 1000)
sampling_method: Whether you want to sample by "rank" or "diversity" (default:"diversity")
column_name: Name of the column in the csv with filenames (default:file_url.csv)
main.py --config ../configs/main_config.yml --clean_speech_classifier ../models/classifiers/labeler.onnx --sig_dnsmos_path ../models/dns835/sig.onnx --bak_ovr_dnsmos_path ../models/dns835/bak_ovr.onnx --noise_type_classifier_path ../models/noise_type_model/tagger_236.onnx --centroids ../models/models/centroids/centroids.npz --ood_centroids ../models/centroids/centroids_dns_train.npz --save_dir ./outputs
If you use our work, please cite using the following bibtex:
@misc{gitiaux2021aura,
title={Aura: Privacy-preserving augmentation to improve test set diversity in noise suppression applications},
author={Xavier Gitiaux and Aditya Khant and Chandan Reddy and Jayant Gupchup and Ross Cutler},
year={2021},
eprint={2110.04391},
archivePrefix={arXiv},
primaryClass={eess.AS}
}
@article{reddy2021dnsmos,
title={DNSMOS P. 835: A non-intrusive perceptual objective speech quality metric to evaluate noise suppressors},
author={Reddy, Chandan KA and Gopal, Vishak and Cutler, Ross},
journal={arXiv preprint arXiv:2110.01763},
year={2021}
}
https://github.com/microsoft/DNS-Challenge/tree/master/DNSMOS
This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.opensource.microsoft.com.
When you submit a pull request, a CLA bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA.
This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact [email protected] with any additional questions or comments.
This project may contain trademarks or logos for projects, products, or services. Authorized use of Microsoft trademarks or logos is subject to and must follow Microsoft's Trademark & Brand Guidelines. Use of Microsoft trademarks or logos in modified versions of this project must not cause confusion or imply Microsoft sponsorship. Any use of third-party trademarks or logos are subject to those third-party's policies.