A Three-Phase Foundation Model for Tax-Aware Personalized Portfolio Management

📅 2026-06-29
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
This work addresses three key limitations in existing financial reinforcement learning—asset lock-in, single-objective optimization, and static user modeling—by proposing a novel three-stage deep reinforcement learning framework. The approach introduces Chronos, a time-series foundation model, into portfolio management, leveraging self-supervised pretraining to learn asset-agnostic representations. It incorporates a goal-conditioned reward mechanism and an intent-routing strategy within a Mixture-of-Experts (MoE) architecture to jointly optimize six distinct investment objectives while mitigating gradient conflicts. Furthermore, personalized investment goals are dynamically inferred from real trading behavior through LoRA-based fine-tuning and natural language parsing, eliminating the need for user questionnaires. The method achieves zero-shot generalization across arbitrary publicly traded assets and demonstrates significant improvements in real-world investment performance.
📝 Abstract
We present a three-phase deep reinforcement learning system for personalized portfolio management that addresses three limitations shared by all prior financial RL work: 1) ticker lock-in, 2) monolithic objectives , and 3) static user models. Phase 1 pretrains a ticker-identity-free cross asset encoder via self-supervised learning on a multi-asset corpus, augmented by a frozen parallel branch using Chronos, a T5-based time series foundation model, fused via a learned gating mechanism. To our knowledge, this is the first application of a time series foundation model to portfolio management RL. The encoder generalizes to any publicly traded asset via a 50-dimensional observable metadata vector that requires no retraining for new tickers. Phase 2 fine-tunes a MoE (Mixture of Experts) portfolio actor critic with PPO under an objective-conditioned reward that simultaneously serves six distinct investment goals sampled per episode: short-term alpha, short-term gain, long-term gain, capital preservation, tax-loss harvesting, and long-term-gains-only. A MoE architecture assigns each objective to a specialized expert head (momentum, growth, defensive, tax-aware), and a learned intent router blends experts based on the active objective and current market regime, which eliminates cross-objective gradient conflict. Phase 3 adds a lightweight personalization layer further adapted at inference time to each individual via a 76-parameter LoRA module fine-tuned on real brokerage transaction history, inferring investment objectives from revealed trading behavior rather than questionnaires. A natural language intent parser converts free-form goals directly into structured investment objective parameters.
Problem

Research questions and friction points this paper is trying to address.

ticker lock-in
monolithic objectives
static user models
personalized portfolio management
tax-aware investment
Innovation

Methods, ideas, or system contributions that make the work stand out.

foundation model
tax-aware portfolio management
mixture of experts
self-supervised learning
personalization
🔎 Similar Papers
No similar papers found.