value aware model learning