02 / Research & writing

Publications.

Research publications by Dane Malenfant on agentic and multi-agent reinforcement learning, reciprocity, memory, continual learning, generative modelling, and AI governance.

Research programmes

Questions that connect the papers

My work spans several publication venues, but it is organized here by the research problems that connect it: cooperation among learning agents, memory and credit across time and structure, and the wider social and cultural contexts of AI.

Research programme · Ongoing

Reciprocity & Multi-Agent Learning

How can adaptive agents learn to cooperate when benefits are delayed, hidden, costly to produce, or mediated by another learner?

Accepted Code available

RLC 2026 · Montréal, Canada · August 2026

The Challenge of Hidden Gifts in Multi-Agent Reinforcement Learning

Dane Malenfant and Blake Aaron Richards

Question
How can agents learn reciprocal behaviour when helpful actions produce delayed or difficult-to-observe benefits?
Contribution
We introduce a shared-key grid-world that makes helpful actions unobservable, show that several state-of-the-art multi-agent reinforcement learning algorithms fail to obtain the collective reward, and derive a learning-aware policy-gradient correction that reduces variance and improves reliable convergence to collective success.
Variance in collective success across training for four policy-gradient variants
Figure 5b. Variance in collective success across training for vanilla policy-gradient, maximum-entropy, correction, and self-correction agents. Self-correction produces the lowest variance, making cooperative learning more reliable.
Preprint Revised v7 Code available

arXiv v7 · Revised September 2026 · Social Simulation with LLMs @ COLM 2026

Moral Hazard in Multi-Agent Language Models

Dane Malenfant

Question
How do language agents cooperate when gathering useful information is costly, hidden, and primarily benefits another agent?
Contribution
I introduce the Dialogue Moral Hazard Game and evaluate fourteen open-weight and four frontier models across acquisition, communication, downstream use, and team success. I also introduce CREDIT, a counterfactual mechanism-aligned prompt optimizer that rewards robust causal contribution and preserves query-mediated behaviour across changing information structures.
GPT-5.6 Sol query, team-success, and local-reward rates across query costs
Figure 2A. GPT-5.6 Sol's query-cost sweep. As querying becomes more expensive, querying and team success decline while preservation of local reward increases.
Accepted Revised v2 Code available

Accepted to Continual World Models Workshop · NeurIPS 2026 · arXiv v2, revised August 2026 · Previously presented at World Modeling Workshop 2026

Reinforcing the World's Edge: Continual Learning at the Multi-Agent–World Boundary

Dane Malenfant

Question
How long can success-conditioned reusable structure remain valid when peer learning changes the focal agent's induced rewards and dynamics?
Contribution
I recast decentralized peer learning as an agent-centric continual-RL problem and define an invariant core of high-coverage abstract patterns in successful trajectories. A worst-case-tight stability theorem yields Ω(1/η) survival and conditional Θ(1/η) first-exit laws; registered 64-stream continual-control and cue-MNIST studies show that core erosion predicts failure and enables near-oracle intervention.
Mean reward comparison for core-triggered, oracle, random-matched, never-switch, and return-trigger control
Figure 2c. In the registered 64-stream continual-control confirmation, core-triggered control achieves mean reward 0.994, matching the oracle and outperforming random-matched (0.536), never-switch (0.460), and return-trigger (0.458) controls.

Research programme · Ongoing

Credit Assignment, Memory & Compositional Learning

How can learning systems isolate, retain, and reuse the information that matters across long horizons or unseen combinations?

Preprint arXiv v1

arXiv v1 · Submitted September 2026

KV-streams for Efficient Compaction in Agentic Reinforcement Learning

Emiliano Penaloza, Dane Malenfant, Dheeraj Vattikonda, Roger Creus Castanyer, Siddarth Venkatraman, Abhay Puri, Jonathan Light, Matthew James Sargent, Augustine N. Mavor-Parker, Massimo Caccia, Lucas Caccia, Glen Berseth, Esmeralda S. Whitammer, Alessandro Sordoni, Minseon Kim, Marc-Alexandre Côté, Laurent Charlin, and Guillaume Lajoie

Question
How can agentic language models compact increasingly long interaction traces without repeatedly prefilling the full context or losing information needed later?
Contribution
We introduce KV-streams, a plug-and-play mechanism that streams the KV cache across compaction events instead of flushing it. Across three compaction strategies, it delivers 2.6–5× wall-clock training speedups without evidence of performance loss, while the streamed cache can learn through reinforcement learning alone to act as recurrent state carrying information no longer present in context.
Five actor-name candidates and the actor recalled after context eviction
Detail from Figure 8B. After the assigned actor name leaves the visible context, the model must recover it from the candidates using information carried by retained KV-cache entries. This tests whether KV-streams trained on fruit assignments generalizes to actor names.
Published Code available

NeurIPS 2023 · New Orleans, USA · December 2023

Contrastive Retrospection: honing in on critical steps for rapid learning and generalization in RL

Chen Sun, Wannan Yang, Thomas Jiralerspong, Dane Malenfant, Benjamin Alsbury-Nealy, Yoshua Bengio, and Blake Aaron Richards

Question
How can reinforcement-learning agents identify the few temporally distant decisions that determine eventual success?
Contribution
We introduce Contrastive Retrospection (ConSpec), a plug-in method that uses offline contrastive learning to discover interpretable prototypes of critical states and turn them into intrinsic rewards. ConSpec improves learning across diverse tasks and supports out-of-distribution generalization when sensory features change.
Reward curves for ConSpec with PPO and three reinforcement-learning baselines
Figure 2b. Reward on the 3D OrangeTree task. ConSpec with PPO approaches the maximum reward, while PPO, SynthRs with PPO, and CURL with PPO remain substantially lower.
Spotlight Workshop paper

FoGen @ ICML 2026 · Seoul, South Korea · July 2026

Compositional Flow Matching with Factored Velocity Fields

Avery Hee-Woon Ryoo, Dane Malenfant, Matthew G. Perich, and Guillaume Lajoie

Question
How can conditional generative models generalize to novel combinations of factors that were absent from training?
Contribution
We propose a factored conditional flow-matching architecture with a shared base velocity and factor-specific heads. It matches or outperforms a monolithic baseline on zero-shot combinations, substantially lowers held-out FID on the hardest Shapes3D splits, and supports adding a new factor without retraining existing components.
Shared encoder-decoder architecture with factor-specific velocity heads
Figure 1. Factored velocity field for zero-shot compositional generalization. A shared encoder-decoder base velocity is augmented by factor-specific conditioned heads whose outputs are combined at the bottleneck.

Interdisciplinary programme

AI in Society & Culture

How do technical systems interact with institutions, public policy, language, and the narratives through which people interpret causality?

Workshop paper Poster available

Harms and Risks of AI in the Military 2024 · Montréal, Canada · December 2024

Inconsistencies in Artificial Intelligence Strategy Alignment of NATO Member States

Itai Epstein, Dane Malenfant, Sara Parker, and Cella Wardrop

Question
How consistently have NATO member states publicly articulated policies for military uses of artificial intelligence?
Contribution
We build a comparative dataset from official national strategies and public statements. Although 88% of NATO members had a national AI strategy, only 34% had published military-specific AI policy, revealing a substantial transparency and alignment gap.
Stacked policy-strategy coverage bars for eleven NATO member states
Derived visualization of public national AI, digital, national security, military-AI, and cybersecurity strategies among the NATO members shown. The complete analysis covers all 32 members, finding national AI strategies for 28 members but military-AI-specific policies for only 11.
Published

Text2Story @ ECIR 2022 · Stavanger, Norway · April 2022

Causality Mining in Fiction

Margaret Meehan, Dane Malenfant, and Andrew Piper

Question
Do causal-relation models developed on standard NLP benchmarks transfer reliably to literary fiction?
Contribution
We introduce a labeled dataset containing 548 positive causal event pairs from 141 contemporary literary passages, together with matched negative examples. Comparing feature-based and BERT models against SemEval data reveals linguistic and domain differences that complicate causal-relation extraction from fiction.
Random Forest feature weights for literary and SemEval causal-relation data
Figure 2. Feature weights from Random Forest models trained on annotated literary and SemEval data. Semantic similarity is the strongest indicator of causal relations in both datasets, with verb-related features also contributing.