Training dynamics in Physics-Informed Neural Networks with feature mapping