DSWM: Distributional Social World Models for multi-outcome prediction

A 4B parameter model trained via supervised fine-tuning and GRPO predicts multiple plausible future outcomes and their probabilities in a single output. In testing across five environments, the DSWM architecture outperformed existing models on 13 of 15 metrics for social interaction tasks.