Something like this? https://arxiv.org/abs/2406.01579