The interaction between the wheel and rail greatly influences the dynamic response of railway vehicles on the track. A roller rig facility can be used to study and monitor real time parameters that influence wheel-rail interaction such as wear, adhesion, friction and corrugation without actual field tests being carried out. This paper presents the development of the mathematical models for full scale roller rig and 1/5 scale roller rig and the wear prediction model based on KTH wear function. The simulated critical speed for the 1/5 scale roller rig is about one-fifth of the critical speed for the full scale model so the simulated results compare well with the theory related to wheel-rail contact and dynamics. Also the differences between the simulated rolling radii for the full scale model with and without wear function are analysed. This paper presents the initial stage of a large scale research project where the influence of wear on the wheel-rail performance will be studied in more depth.