Speaker diarization with session-level speaker embedding refinement using graph neural networks
This work addresses the challenge of distinguishing speakers in specific meeting sessions for applications like transcription and analysis, representing an incremental improvement over existing methods.
The paper tackles the problem of speaker diarization by refining speaker embeddings locally within each session using graph neural networks, resulting in significantly improved clustering performance and achieving state-of-the-art results on the NIST SRE 2000 CALLHOME database.
Deep speaker embedding models have been commonly used as a building block for speaker diarization systems; however, the speaker embedding model is usually trained according to a global loss defined on the training data, which could be sub-optimal for distinguishing speakers locally in a specific meeting session. In this work we present the first use of graph neural networks (GNNs) for the speaker diarization problem, utilizing a GNN to refine speaker embeddings locally using the structural information between speech segments inside each session. The speaker embeddings extracted by a pre-trained model are remapped into a new embedding space, in which the different speakers within a single session are better separated. The model is trained for linkage prediction in a supervised manner by minimizing the difference between the affinity matrix constructed by the refined embeddings and the ground-truth adjacency matrix. Spectral clustering is then applied on top of the refined embeddings. We show that the clustering performance of the refined speaker embeddings outperforms the original embeddings significantly on both simulated and real meeting data, and our system achieves the state-of-the-art result on the NIST SRE 2000 CALLHOME database.