Yes, this paper says exactly what you talked about: https://arxiv.org/abs/2404.01332