Giter VIP home page Giter VIP logo

ecapa-tdnn's Introduction

ECAPA-TDNN

Unofficial implementation of ECAPA-TDNN

Remarks

image

Training a ECAPA-TDNN model took 3 days.

Effect of Feature Re-scaling

image

image

image

The classification loss is bounded by the rescaling factor s, and the number of classes.

When the loss bound is too large, a model fails to discriminate speakers.

Green - s = 1

Gray - s = 16

image

Attention Matrix Alpha

image

  1. Channel independent attention values (Alpha Matrix) are mostly similar along channel axis, indicating channel independent attention is unnecessary.
  2. Mean attention score (Mean Alpha) tends to be high when the voice is present, indicating the model is "listening" to what is being spoken. The attention score seems pretty similar to VAD(voice activity detection) score.
  3. Attention score is not correlated to the volume of the speech. The model sometimes focuses on how the speaker ends the speech (the maximum value on the graph). It would be interesting to investigate which portion of the speech characterizes a distinctive speaker.

Speaker Embedding

image

Cosine similarity of speaker embedding is being close to zero during the training proces, meaning AAM (Additive Angular Margin) Loss is successful in diagonalizing the speaker identity.

Inference Similarity

inference_result

Inference result seems to be discriminitive, but there was a mistake. An identity vector of a speaker is calculated as a mean of the vectors from all speechs of that speaker. As seens from below histogram, even for completely random vectors h(x) and h(y), cosine similarity between h(x) and mu(x) becomes biased.

bias

bias_equation

Cosine Simarity and Dimension

As dimension h increases, distribution of cosine similarity between two random vector becomes a normal distribution lower standard deviation. In other words, two random vectors are more likely to be diagonal as dimension increases.

h_and_cos

References

  1. ECAPA-TDNN: Emphasized Channel Attention, Propagation and Aggregation in TDNN Based Speaker Verification
  2. ArcFace: Additive Angular Margin Loss for Deep Face Recognition
  3. VoxCeleb

TODO

  • Get VoxCeleb statistics (file length and counts, sampling rate)

  • Build speaker tokenizer

  • Apply VAD to segment audio files

  • Run model on VoxCeleb1

  • Implement .m4a loading

  • Run model on VoxCeleb2

ecapa-tdnn's People

Contributors

joovvhan avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

Forkers

forwiat ts0923

ecapa-tdnn's Issues

What's the test EER of VoxCeleb1

Hi, Joovvhan!
Thank you for your excellent code!
I see official ECAPA-TDNN can get test EER 1.01% on VoxCeleb1, what's the test EER by your code? Thanks!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.