DeepFilterNet3 — MLX

MLX-compatible weights for DeepFilterNet3, a real-time speech enhancement model that suppresses background noise from audio.

This is a direct conversion of the original PyTorch weights to safetensors format for use with MLX on Apple Silicon.

Origin

• Original model: DeepFilterNet3 by Hendrik Schröter
• Paper: DeepFilterNet: Perceptually Motivated Real-Time Speech Enhancement
• License: MIT (same as the original)
• Conversion: PyTorch → safetensors via the included convert_deepfilternet.py script

No fine-tuning or quantisation was applied — the weights are numerically identical to the original checkpoint.

Files

Model Details

Usage

Swift (mlx-audio-swift)

import MLXAudioSTS

let model = try await DeepFilterNetModel.fromPretrained("iky1e/DeepFilterNet3-MLX")
let enhanced = try model.enhance(audioArray)

Python (mlx-audio)

from mlx_audio.sts.models.deepfilternet import DeepFilterNetModel

model = DeepFilterNetModel.from_pretrained(version=3, model_dir="path/to/local/dir")
enhanced = model.enhance("noisy.wav")

Converting from PyTorch

To re-create this conversion from the original DeepFilterNet checkpoint:

python convert_deepfilternet.py \
  --input /path/to/DeepFilterNet3 \
  --output ./DeepFilterNet3-MLX \
  --name DeepFilterNet3

The input directory should contain a config.ini and a checkpoints/ folder from the original repo.

Citation

@inproceedings{schroeter2023deepfilternet3,
  title={DeepFilterNet: Perceptually Motivated Real-Time Speech Enhancement},
  author={Schr{\"o}ter, Hendrik and Rosenkranz, Tobias and Escalante-B., Alberto N. and Maier, Andreas},
  booktitle={INTERSPEECH},
  year={2023}
}