Brain-to-text decoding with context-aware neural representations and large language models.

Abstract

<i>Objective</i>. Decoding attempted speech from neural activity offers a promising avenue for restoring communication abilities in individuals with speech impairments. Previous studies have focused on mapping neural activity to text using phonemes as the intermediate target. While successful, decoding neural activity directly to phonemes ignores the context dependent nature of the neural activity-to-phoneme mapping in the brain, leading to suboptimal decoding performance.<i>Approach</i>. In this work, we propose the use of diphone-an acoustic representation that captures the transitions between two phonemes-as the context-aware modeling target. We integrate diphones into existing phoneme decoding frameworks through a novel divide-and-conquer strategy in which we model the phoneme distribution by marginalizing over the diphone distribution. Our approach effectively leverages the enhanced context-aware representation of diphones while preserving the manageable class size of phonemes, a key factor in simplifying the subsequent phoneme-to-text conversion task.<i>Main results</i>. We demonstrate the effectiveness of our approach on the Brain-to-Text 2024 benchmark, where it achieves state-of-the-art phoneme error rate (PER) of 15.34% compared to 16.62% PER of monophone-based decoding. When coupled with finetuned large language models (LLMs), our method yields a Word error rate (WER) of 5.77%, significantly outperforming the 8.93% WER of the leading method in the benchmark.<i>Significance</i>. These results demonstrate the effectiveness of leveraging context-aware neural representations and LLMs for brain-to-text decoding, thereby expanding the capabilities of speech neuroprostheses and paving the way toward restoring communication in individuals with speech impairments.