AI-based data management