A method for machine learning based prediction of at least one subsequent UI layout is provided. The method may include detecting, by the electronic device, a first transition event. Further, the method may include identifying, by the electronic device, a UI layout associated with a first application of the electronic device. Further, the method may include predicting, by the electronic device, the at least one subsequent UI layout to be displayed based on at least one transition parameter, wherein the at least one subsequent UI layout is associated with at least one of the first application or at least one second application. Further, the method may include loading, by the electronic device, the at least one subsequent UI layout in a memory of the electronic device.