All units in game can have innate specialization (except maybe the champions). For example:
Warrior: medium health, medium defence, medium attack
Mage: low health, low defence, high spell mastery, high spell resistance, bonus to fire/cold/lighting attack
Assassin: low health, low defence, high attack, high critical chance, high dodge
Defender: high health, high defence, low attack
Each unit has base stats according to its specialization. When unit gains a new level, its stats are increased according to its specialization.
This idea can improve trained units so they can compete with champions.