The present disclosure provides a method for wave propagation prediction based on a 3D ray tracing engine and machine-learning based dominant ray selection. The method includes receiving, integrating, and processing input data. Integrating and processing the input data includes dividing a cone of the original millimeter wave (mmWave) into a plurality of sub cones; determining a contribution weight of rays coming from each sub cone to the received signal strength (RSS) at a receiving end of interest; and determining rays coming from one or more sub cones that have a total contribution weight to the RSS larger than a preset threshold value as dominant rays using a neural network obtained through a machine learning approach. The method further includes performing ray tracing based on the input data and the dominant rays to predict wave propagation.