latentbrief
← Back to news
Launch1mo ago

Custom Reward Functions for Multi-Turn Reinforcement Learning with Amazon Nova Forge

AWS ML Blog2 min brief

In brief

  • Amazon's Nova Forge introduces a new approach to multi-turn reinforcement learning (RL), focusing on custom reward functions that guide AI models to learn correctly over extended interactions.
  • Traditional RL can fail silently even when training looks good because subtle errors in reward design lead the model astray.
  • Nova Forge addresses this by allowing users to define precise reward criteria, ensuring models behave as intended across multiple steps-like calling tools or recovering from mistakes.
  • The key innovation is Reinforcement Fine-Tuning (RFT), which differs from standard methods by learning from evaluation signals on the model's outputs instead of relying on curated examples.
    • This makes it easier for developers to train models iteratively through feedback, rather than needing detailed reasoning paths.
  • Multi-turn RFT optimizes cumulative rewards across sequences of actions, leading to better generalization and performance in diverse tasks.
  • For teams without the infrastructure to manage environments, Nova Forge offers a serverless multi-turn RL option.
    • This feature is now generally available, providing flexibility for different user needs.
  • Developers should focus on defining clear reward functions while Nova Forge handles coordination during training.
  • The approach emphasizes safety by executing model-generated code securely within reward calculations and monitoring each component's contribution to ensure reliable learning outcomes.
  • Looking ahead, this development could significantly improve the reliability of AI systems in complex, multi-step tasks-where subtle design choices can make or break performance.

Terms in this brief

Reinforcement Fine-Tuning
A method where AI models learn by adjusting their behavior based on feedback from evaluations of their outputs. Unlike traditional methods that rely on pre-curated examples, RFT allows for iterative training through user feedback, making it easier to refine model performance without needing extensive predefined reasoning paths.

Read full story at AWS ML Blog →

More briefs