Downloads · 30 days
11.6K
16% of all-time downloads
nvidia/RE-USE
RE-USE is a audio-to-audio model from nvidia. Use it for the audio-to-audio task on the model card, and read the license before you ship it in a product. It is set up for mamba-ssm. The card lists the license as other.
In universal speech enhancement, the goal is to restore the quality of diverse degraded speech while preserving fidelity, ensuring that all other factors remain unchanged, e.g., linguistic content, speaker identity, e…
Downloads · 30 days
11.6K
16% of all-time downloads
All-time downloads
70.3K
Public
Parameters
9.6M
78 MB on disk
Likes
118
Trending 4
Click a slice to open those files.
.safetensors38.6 MB · 98%
From the Hugging Face model README
In universal speech enhancement, the goal is to restore the quality of diverse degraded speech while preserving fidelity, ensuring that all other factors remain unchanged, e.g., linguistic content, speaker identity, emotion, accent, and other paralinguistic attributes. Inspired by the distortion–perception trade-off theory, our proposed single model achieves a good balance between these two objectives and has the following desirable properties:
This model is for research and development only.
Directly try our Gradio Interactive Demo by uploading your noisy audio/video !!
(For Mamba setup)Pre-built Docker environments can be downloaded here to simplify Mamba setup.
If you need bandwidth extension:
pip install resampy
huggingface-cli download nvidia/RE-USE --local-dir ./REUSE --local-dir-use-symlinks False
cd ./REUSE
Follow the simple steps below to generate enhanced speech using our model:
noisy_audio/sh inference.sh
enhanced_audio/.That's all !
Note:
a. You can enable bandwidth extension by setting the target bandwidth using the BWE argument in the script.
If your noisy speech files are long and may cause GPU out-of-memory (OOM) errors, please use the following procedure instead:
long_noisy_audio/sh inference_chunk.sh
Long_enhanced_audio/.Note:
a. You can enable bandwidth extension by setting the target bandwidth using the BWE argument in the script.
b. You can also configure the chunk_size_in_seconds and hop_length_portion directly in the script.
This model is released under the NVIDIA One-Way Noncommercial License (NSCLv1).
Global.
Researchers and general users can use this model to enhance the quality of their speech data.
Hugging Face 2026/03/18
[1] Rethinking Training Targets, Architectures and Data Quality for Universal Speech Enhancement, 2025. (Note: The released model checkpoint differs from the one reported in the paper. It incorporates additional degradation types (e.g., microphone response and more codecs) and is fine-tuned on a smaller, high-quality clean subset.)
Architecture Type: Convolutional encoder, Convolutional decoder, and Mamba for time–frequency modeling <br> Network Architecture: Bi-directional Mamba with 30 layers <br> Number of model parameters: 9.6M <br>
Input Type(s): Audio <br> Input Format(s): .wav files <br> Input Parameters: One-Dimensional (1D) <br> Other Properties Related to Input: 8000 Hz - 48000 Hz Mono-channel Audio <br>
Output Type(s): Audio <br> Output Format: .wav files <br> Output Parameters: One-Dimensional (1D) <br> Other Properties Related to Output: 8000 Hz - 48000 Hz Mono-channel Audio <br>
Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA’s hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions.
Runtime Engine(s):
Supported Hardware Microarchitecture Compatibility:
Preferred Operating System(s):
The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
Current version: 30USEMamba_peak+GAN_tel_mic_1134k
Data Modality: Audio
Audio Training Data Size: Less than 10,000 Hours
Acceleration Engine: None <br> Test Hardware: NVIDIA A100
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse. Please report model quality, risk, security vulnerabilities or NVIDIA AI Concerns here.
Please consider to cite our paper and this framework, if they are helpful in your research.
@article{fu2026rethinking,
title={Rethinking Training Targets, Architectures and Data Quality for Universal Speech Enhancement},
author={Fu, Szu-Wei and Chao, Rong and Yang, Xuesong and Huang, Sung-Feng and Zezario, Ryandhimas E and Nasretdinov, Rauf and Juki{\'c}, Ante and Tsao, Yu and Wang, Yu-Chiang Frank},
journal={arXiv preprint arXiv:2603.02641},
year={2026}
}