A machine learning based approach towards high-dimensional mediation analysis