Skip to content

Aryan Dutt / SINGAPORE

Available for research and engineering opportunities

ML

Aug 2025 – Dec 2025

Federated Whisper Aggregation Pipeline

Fine-tuning a production speech model across decentralised datasets without moving any data.

With Vincenzo and Manfred

Problem

Speech datasets are often too sensitive to centralise, which rules out conventional fine-tuning. Federated training removes the data-sharing requirement but introduces the question of which aggregation strategy actually preserves model quality.

Approach

  • 01Engineered a federated pipeline fine-tuning whisper-large-v3-turbo across decentralised speech datasets using Flower on a Linux VM.
  • 02Implemented weighted checkpoint averaging across more than ten aggregation strategies, with automated orchestration across configurations.
  • 03Evaluated on HuggingFace datasets, logging WER and BLEU per strategy.
  • 04Validated exported state dicts directly against Whisper's native inference backend.

Outcome

A reproducible aggregation and evaluation stack that compares strategies under zero data sharing.

Stack

  • Python
  • Flower
  • PyTorch
  • HuggingFace
  • Whisper