Design a data imputation strategy with AI