zuazo / megconformer-speech-detection

MEGConformer for Speech Detection

Conformer-based MEG decoder for binary speech detection, trained with 10 different random seeds for reproducibility.

Model Performance

• Holdout score of seed 0: 88.90%

Quick Start

Load Best Model

import torch
from huggingface_hub import hf_hub_download

from libribrain_experiments.models.configurable_modules.classification_module import (
    ClassificationModule,
)

# Download a checkpoint (seed-0)
checkpoint_path = hf_hub_download(
    repo_id="zuazo/megconformer-speech-detection", filename="seed-0/pytorch_model.ckpt"
)

# Choose device
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# Load model and move to device
model = ClassificationModule.load_from_checkpoint(checkpoint_path, map_location=device)
model.eval()

# Inference
meg_signal = torch.randn(1, 306, 125, device=device)  # Create directly on device

with torch.no_grad():
    logits = model(meg_signal)
    prediction = torch.argmax(logits, dim=1)  # 0=silence, 1=speech

print(f"Prediction: {'Speech' if prediction.item() == 1 else 'Silence'}")

Model Details

• Architecture : Conformer Small
  • Hidden size: 144
  • FFN dim: 576
  • Layers: 16
  • Attention heads: 4
  • Depthwise conv kernel: 31
• Input : 306-channel MEG signals
• Window size : 2.5 seconds (625 samples at 250 Hz)
• Output : Binary classification (silence/speech)
• Training : LibriBrain 2025 Standard track

Reproducibility

All 10 random seeds are provided to ensure reproducibility.

Citation

@misc{dezuazo2025megconformerconformerbasedmegdecoder,
      title={MEGConformer: Conformer-Based MEG Decoder for Robust Speech and Phoneme Classification}, 
      author={Xabier de Zuazo and Ibon Saratxaga and Eva Navas},
      year={2025},
      eprint={2512.01443},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2512.01443}, 
}

License

The 3-Clause BSD License

Links

• Paper : arXiv:2512.01443
• Code : GitHub
• Competition : LibriBrain 2025