AI-driven reconstruction