Learning Compact Features via In-Training Representation Alignment