elyxlz/team17

Python

0

32 commits

updated Nov 23, 2024

See the code

README

Seren

Seren is a 100% local and custom speech2speech pipeline purpose built for emotional understanding and therapeutic conversations. ๐Ÿ—ฃ๏ธ

Architecture and Training

The system operates as a 2-stage pipeline:

  1. Audio Language Model (ALM) ๐Ÿง : A LLaMA base model trained to directly process Whisper embeddings in its context, bypassing traditional transcription. By feeding emotional and tonal information directly from speech embeddings into the model, we maintain crucial affective signals that are typically lost in text intermediates - essential for therapeutic conversations.

  2. Text-to-Speech ๐Ÿ”Š: Powered by T5-TTS for voice synthesis.

The training data consisted of hand-curated therapeutic conversations, collected through a distributed yt-dlp + ffmpeg scraping setup. Our preprocessing pipeline, implemented with distributed PyTorch DataLoaders, performed:

  • Whisper transcription
  • Pyannote speaker diarization
  • wav2vec 2.0 forced alignment timestamping
  • Whisper embedding generation
  • User/therapist identification based on first/second person pronoun ratios

The ALM was initialized from Ultravox weights and fine-tuned using LoRA and activation checkpointing via a custom trainer.

The TTS stage currently represents our primary latency bottleneck. Our planned evolution involves interleaving mimi (Moshi codec) and text tokens (thought tokens) in the assistant outputs to create a unified end-to-end speechLM. โšก

Contributors

elyxlz

15 commits

julien-blanchon

11 commits

aaftall

5 commits

em-jaouen

1 commits

elyxlz/team17

Python

0

32 commits

updated Nov 23, 2024

See the code

README

Seren

Seren is a 100% local and custom speech2speech pipeline purpose built for emotional understanding and therapeutic conversations. ๐Ÿ—ฃ๏ธ

Architecture and Training

The system operates as a 2-stage pipeline:

  1. Audio Language Model (ALM) ๐Ÿง : A LLaMA base model trained to directly process Whisper embeddings in its context, bypassing traditional transcription. By feeding emotional and tonal information directly from speech embeddings into the model, we maintain crucial affective signals that are typically lost in text intermediates - essential for therapeutic conversations.

  2. Text-to-Speech ๐Ÿ”Š: Powered by T5-TTS for voice synthesis.

The training data consisted of hand-curated therapeutic conversations, collected through a distributed yt-dlp + ffmpeg scraping setup. Our preprocessing pipeline, implemented with distributed PyTorch DataLoaders, performed:

  • Whisper transcription
  • Pyannote speaker diarization
  • wav2vec 2.0 forced alignment timestamping
  • Whisper embedding generation
  • User/therapist identification based on first/second person pronoun ratios

The ALM was initialized from Ultravox weights and fine-tuned using LoRA and activation checkpointing via a custom trainer.

The TTS stage currently represents our primary latency bottleneck. Our planned evolution involves interleaving mimi (Moshi codec) and text tokens (thought tokens) in the assistant outputs to create a unified end-to-end speechLM. โšก

Contributors

elyxlz

15 commits

julien-blanchon

11 commits

aaftall

5 commits

em-jaouen

1 commits

Languages

Python

53.6%

Jupyter Notebook

21.6%

JavaScript

16.1%

TypeScript

4.7%

Rust

2.5%