Trait-Aware Policy Optimization for Autoregressive Multi-Trait Essay Scoring

Zhengyang Wang, Sanwoo Lee, Jiaxin Wang, Chenxi Miao, Weikang Li +1 more
5/25/2026
cs.CL

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

Multi-trait essay scoring aims to provide fine-grained evaluation of writing quality across multiple dimensions. However, how to effectively post-train autoregressive scoring models remains underexplored. In this paper, we propose Trait-Aware Policy Optimization (TAPO), a post-training framework tailored to autoregressive multi-trait scoring. Our method decomposes rewards along both the sample and trait dimensions, combining global scoring consistency, trait-level accuracy, format validity, and inter-trait dependency preservation. In addition, we use enhanced prompts throughout training by incorporating original prompt texts and trait descriptions, providing richer semantic information for trait-specific score generation. Experiments across multiple backbone models show that our method consistently improves multi-trait scoring performance over supervised fine-tuning and scalar-reward optimization baselines, demonstrating the effectiveness and transferability of trait-aware post-training for essay scoring.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{wang2026traitaware,
  title  = {Trait-Aware Policy Optimization for Autoregressive Multi-Trait Essay Scoring},
  author = {Zhengyang Wang and Sanwoo Lee and Jiaxin Wang and Chenxi Miao and Weikang Li and Yunfang Wu},
  year   = {2026},
  eprint = {2605.25731},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2605.25731v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2605.25731.svg)](https://codeofpaper.com/paper/2605.25731)

Discussion