AI for data processing