PERSA: Reinforcement Learning for Professor-Style Personalized Feedback with LLMs

· Source: Paper Index on ACL Anthology · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Emerging Technologies & Innovation · Depth: Expert, quick

Summary

PERSA is a Reinforcement Learning from Human Feedback (RLHF) pipeline designed to adapt large language models (LLMs) for generating personalized, professor-style programming feedback. It addresses the challenge of aligning an LLM's stylistic tone with a specific instructor while maintaining diagnostic correctness. PERSA combines supervised fine-tuning on professor demonstrations, reward modeling from pairwise preferences, and Proximal-based policy optimization. Critically, it employs parameter-efficient fine-tuning, updating only the top transformer blocks and their feed-forward projections to minimize global parameter drift and increase stylistic controllability. Evaluated on APPS, PyFiXV, and CodeReviewQA benchmarks using Llama-3 and Gemma-2 backbones, PERSA achieved a Style Alignment Score (SAC) of 96.2% (from 34.8% for Base) and 100% Correctness Accuracy (CA) on APPS, demonstrating strong style transfer without sacrificing correctness.

Key takeaway

For Machine Learning Engineers developing educational AI tools, PERSA offers a practical method to align LLM output with specific pedagogical styles. You can achieve high style alignment, boosting Style Alignment Score to 96.2% while maintaining 100% Correctness Accuracy, by implementing parameter-efficient RLHF techniques. Consider applying similar constrained fine-tuning approaches to customize feedback tone without compromising factual accuracy in your applications.

Key insights

PERSA uses RLHF and parameter-efficient fine-tuning to align LLM feedback style with instructors while preserving correctness.

Principles

Method

PERSA's RLHF pipeline integrates supervised fine-tuning on professor demonstrations, reward modeling from pairwise preferences, and Proximal-based policy optimization, focusing parameter updates on top transformer blocks.

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.