Is this paper achieving what you mean: https://arxiv.org/pdf/2311.16728.pdf

Caveat, this field is far outside my wheelhouse. So if not, I'd love to understand more about the nuances here.