arXiv:2605.31463: PithTrain, kompaktni sustav za treniranje MoE modela prilagođen agentima

Rad arXiv:2605.31463 predstavlja PithTrain, kompaktni sustav za treniranje Mixture-of-Experts (MoE) modela osmišljen tako da bude prilagođen AI agentima za kodiranje. Autori tvrde da postojeći produkcijski okviri kriju skrivene troškove kada ih koriste agenti, što ova studija pokušava izmjeriti i smanjiti.

Što je Mixture-of-Experts?

Mixture-of-Experts je arhitektura modela u kojoj je mreža podijeljena na više specijaliziranih “eksperata”, a za svaki ulaz se aktivira samo dio njih. To omogućuje velike modele uz manji trošak izračuna po upitu. Treniranje MoE modela je složeno jer zahtijeva pažljivo usmjeravanje podataka prema pravim ekspertima i učinkovito raspoređivanje rada po GPU-ima.

Što je agent-task efficiency?

Rad uvodi novu metriku agent-task efficiency (ATE), definiranu kao “trošak korištenja agenata za kodiranje da razumiju, koriste i prošire neki framework”. Autori ističu da se dosadašnje evaluacije fokusiraju isključivo na propusnost (throughput), zanemarujući koliko je teško agentu raditi s danim sustavom. Za mjerenje toga uveden je ATE-Bench, koji pokriva stvarne zadatke vezane uz frameworke za treniranje.

Kakvi su rezultati?

PithTrain zadržava propusnost usporedivu s produkcijskim okvirima, ali na ATE-Benchu donosi znatne uštede. Postiže do 62 posto manje “Agent Turns” (agentskih koraka potrebnih da agent obavi zadatak) i do 64 posto manje “Active GPU Time” (aktivnog vremena rada grafičkih procesora). Sustav je izgrađen na četiri agentima prilagođena načela dizajna.

Zašto je to važno?

Kako AI agenti sve više preuzimaju zadatke održavanja i proširivanja sustava za treniranje, troškovi koje ti agenti generiraju postaju mjerljiv faktor. PithTrain pokazuje da se framework može optimizirati ne samo za brzinu modela, nego i za lakoću kojom ga agenti razumiju i mijenjaju, što smanjuje i broj koraka i utrošak skupih GPU resursa.