Mame Diarra Toure.

PhD Candidate in Applied Mathematics, McGill University

Open to full-time research roles, starting Fall 2027

I am a PhD candidate in Applied Mathematics at McGill University, advised by David Stephens. My research focuses on Bayesian Deep Learning and Uncertainty Quantification. I expect to complete my PhD in Spring 2027.

I develop scalable variational methods to make neural networks robust, interpretable, and computationally efficient, bridging theoretical rigor and practical deployment to build safe and trustworthy AI systems.

In 2024 I was awarded the Women in AI Excellence Scholarship from Mila (Quebec AI Institute). My academic foundation was built in France: after the intensive Classes PrĂ©paratoires aux Grandes Écoles, I earned an Engineering Degree from ENSIIE and an MSc in Quantitative Finance from Paris-Saclay University, where I graduated with First Class Honours and received the Sophie Germain Excellence Scholarship from the Jacques Hadamard Mathematics Foundation (FMJH). Before my PhD, I worked across quantitative research and applied AI, with roles at BNP Paribas, SociĂ©tĂ© GĂ©nĂ©rale, and CrĂ©dit Agricole, and as a research scientist at JACOBB (Centre for Applied AI).

01 Research & Publications

My primary research develops scalable low-rank variational inference methods (W ≈ ABT) for Bayesian neural networks, with provable guarantees and an emphasis on parameter efficiency. I also study whether these structures let networks automatically adapt their complexity to the data generating process (Occam's Razor).

ICML 2026
Seoul, July 2026
Singular Bayesian Neural Networks Accepted
Mame Diarra Toure, David A. Stephens
International Conference on Machine Learning (ICML), 2026

A low-rank variational inference framework for Bayesian neural networks that parameterizes weights as W = ABT, reducing parameter complexity from O(mn) to O((m+n)r). This induces a posterior singular with respect to the Lebesgue measure, capturing structured weight correlations through shared latent factors. We derive PAC-Bayes generalization bounds and prove loss bounds via the Eckart-Young-Mirsky theorem. Empirically, the method matches 5-member Deep Ensembles with up to 15× fewer parameters and substantially improves out-of-distribution detection across MLPs, LSTMs, and Transformers.

Preprint
2026
Not Just How Much, But Where: Decomposing Epistemic Uncertainty into Per-Class Contributions
Mame Diarra Toure, David A. Stephens
arXiv:2602.21160 (under review), 2026

We decompose mutual information into a per-class uncertainty vector Ck(x) = σk² / (2μk), letting safety-critical classifiers distinguish where a model is uncertain, not just how much. Applied to diabetic retinopathy grading, the critical-class score reduces selective risk by 34.7% over mutual information and 56.2% over variance baselines.

SPIGM @ ICML 2026
Workshop
When Does a Low-Rank Bayesian Neural Network Certify Its Deterministic Center? Accepted
Mame Diarra Toure, David A. Stephens
SPIGM Workshop (Structured Probabilistic Inference & Generative Modeling), ICML 2026

We study when a structured low-rank Gaussian variational posterior can certify a deterministic predictor in a Bayesian neural network with factorized layers Wi = AiBiT. The same model yields three natural certification targets, the posterior Gibbs predictor, the posterior predictive mean, and a deterministic center network; this paper focuses on the deterministic-center route.

PhD Research
Scalable Low-Rank Bayesian Neural Networks
Mame Diarra Toure

A variational inference framework that factorizes weight matrices to reduce parameter complexity from O(n2) to O(n), enabling uncertainty quantification in large-scale models.

Natixis
Master's Project
Combining Neural Networks and Model Diffusion for CVA Pricing
Mame Diarra Toure et al.

Applied deep neural networks to solve high-dimensional PDEs for Credit Valuation Adjustment, overcoming the curse of dimensionality faced by finite-difference methods.

LaMME
Research Internship
Rough Volatility Modelling: Kernel Estimation
Mame Diarra Toure

Implemented statistical techniques to estimate the kernel of Volterra processes, validating the rough-volatility hypothesis in financial time series.

02 Experience

McGill University
Fall 2025, Winter 2026
Graduate Teaching Assistant

TA for MATH 139 (Calculus with Precalculus) and MATH 141 (Calculus 2): leading problem-solving tutorials and strengthening students' conceptual grasp of limits, differentiation, integration, and series.

Crédit Agricole S.A.
Sep 2024 to Jan 2025
Quantitative Model Auditor, Inspection Générale

Designed and executed rigorous evaluation protocols for quantitative and machine learning models across the group, probing robustness, calibration, and governance beyond standard validation. Assessed model sensitivity to input perturbations and distributional shift, and translated findings into remediation recommendations for senior management and group risk committees.

JACOBB (Center for Applied AI)
2022 to 2024
Research Scientist

Applied R&D in NLP and Reinforcement Learning. Built a reinforcement learning agent for advertising budget pacing and a collaborative-filtering matchmaking system for startups, and deployed models from prototype to production.

J.P. Morgan
Summer 2023
Quantitative Research Mentorship Participant

Selected for a competitive mentorship program with in-depth exposure to quantitative research methodologies, financial modeling, and career pathways in global markets.

Société Générale
2021 to 2022
Quantitative Analyst

Contributed to the Haussmann Project: developed and calibrated Probability of Default and Loss Given Default models for low-default portfolios.

BNP Paribas
2021
Quantitative Research Intern

Built mathematical models quantifying the bank's exposure to Central Counterparty Clearing Houses, modeling default-fund contributions with Monte Carlo simulation.

03 Community & Mentorship

I am an active member of Women in AI North America and the Association of Women in Mathematics. I mentor at Jiggen In STEM, supporting high school girls in Senegal, and served as the former representative of AFSA Québec at McGill.