PersonaDrive: Controllable Trajectory Prediction with Multi-Dimensional Driving Personas
Organizations: Kyung Hee University, Yong-in, South Korea · ETRI, Daejeon, South Korea
Abstract
Although recent trajectory prediction and end-to-end autonomous driving methods improve robustness in urban environments, they still lack meaningful controllability. Existing benchmarks either provide no persona-conditioned annotations or support only a single urgency spectrum (i.e., emergency, normal, relaxed), which cannot distinguish personas that share the same urgency level but require different driving dynamics. To address this, we propose (i) the Persona-Conditioned Trajectory (PCT) dataset, which decomposes driving personas along two axes, Temporal Urgency and Ride Comfort, and combines three levels of each to form a grid of nine personas, each paired with natural-language descriptions and trajectories, and (ii) PersonaDrive, a framework that can learn driving personas from language and can generate persona-specific trajectories. PersonaDrive incorporates Persona-Conditioned Anchor Transform (PCAT), which hierarchically reshapes anchors along both axes, and Persona-Conditioned Multi-Modal Fusion (PCMF) for BEV-level persona fusion. Training is supervised by a Hierarchical Guide Loss enforcing axis-aligned physical orderings and an Axis-Decomposed Diversity Loss preventing diagonal mode collapse. Experimental results show that PersonaDrive consistently improves over the compared baselines across multi-dimensional scenarios. The code and PCT dataset are available at https://github.com/VisualAIKHU/PersonaDrive
Figures & tables
| Comf. Low | Comf. Mid | Comf. High | |
| Urg. High | 89.9 | 89.9 | 92.2 |
| Urg. Mid | 98.4 | 82.2 | 82.9 |
| Urg. Low | 94.6 | 73.6 | 67.4 |
| Methods | Avg. ADE | Avg. FDE | Avg. PDMS |
| LTF (TPAMI’22) [ 9 ] | 2.49 | 3.87 | 52.0 |
| Transfuser (TPAMI’22) [ 9 ] | 2.53 | 3.96 | 51.9 |
| UniAD (CVPR’23) [ 18 ] | 2.47 | 3.83 | 52.5 |
| Hydra-MDP (arXiv’24) [ 30 ] | 6.70 | 11.85 | 41.4 |
| PARA-Drive (CVPR’24) [ 54 ] | 4.84 | 9.00 | 51.9 |
| Traj-LLM (T-IV’24) [ 26 ] | 2.66 | 4.20 | 54.1 |
| Method | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| 1D-Urg. | ADE | 3.30 | 2.96 | 3.07 | 2.76 | 2.35 | 2.35 | 2.73 | 1.93 | 2.56 | 2.67 |
| 1D-Comf. | 4.79 | 4.85 | 4.66 | 2.43 | 2.35 | 2.34 | 5.70 | 6.26 | 7.11 | 4.50 | |
| Multi-D | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| 1D-Urg. | FDE | 5.81 | 5.15 | 5.42 | 4.14 | 3.52 | 3.53 | 4.06 | 2.63 | 4.21 | 4.27 |
| 1D-Comf. | 11.76 | 11.40 | 10.32 | 3.69 | 3.52 | 3.50 | 9.84 | 10.88 | 12.87 | 8.64 | |
| Multi-D | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| PCAT | PCMF | Avg. ADE | Avg. FDE | Avg. PDMS | |
| - | - | - | 3.93 | 5.79 | 55.3 |
| ✓ | - | - | 3.52 | 5.43 | 56.0 |
| ✓ | ✓ | - | 3.23 | 4.85 | 56.9 |
| - | ✓ | ✓ | 2.55 | 3.87 | 57.2 |
| ✓ | - | ✓ | 2.61 | 3.97 | 57.5 |
| ✓ | ✓ | ✓ | 2.39 | 3.71 | 57.7 |
| Text Encoders | Avg. ADE | Avg. FDE | Avg. PDMS |
| DeBERTa [ 17 ] | 5.99 | 11.56 | 57.1 |
| RoBERTa [ 37 ] | 4.72 | 8.76 | 62.0 |
| MiniLM [ 51 ] | 2.39 | 3.71 | 57.7 |
| Methods | Avg. ADE | Avg. FDE | Avg. PDMS |
| DiffusionDrive (One-Hot) [ 33 ] | 2.67 | 4.20 | 55.6 |
| PersonaDrive (One-Hot) | 2.54 | 4.01 | 56.6 |
| PersonaDrive (Text) | 2.39 | 3.71 | 57.7 |
| Question | Mean | Std | Median |
| Q1-1 | 4.84 | 0.46 | 5.0 |
| Q1-2 | 4.76 | 0.55 | 5.0 |
| Q1-3 | 4.83 | 0.43 | 5.0 |
| Q1-4 | 4.64 | 0.50 | 5.0 |
| Q1-5 | 4.64 | 0.62 | 5.0 |
| Q1-6 | 4.75 | 0.66 | 5.0 |
| Methods | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| LTF (TPAMI’22) [ 9 ] | ADE | 3.01 | 2.99 | 2.78 | 2.62 | 2.47 | 2.47 | 2.10 | 2.03 | 1.92 | 2.49 |
| Transfuser (TPAMI’22) [ 9 ] | 3.09 | 3.05 | 2.80 | 2.66 | 2.52 | 2.53 | 2.17 | 2.05 | 1.93 | 2.53 | |
| UniAD (CVPR’23) [ 18 ] | 2.99 | 2.98 | 2.76 | 2.53 | 2.46 | 2.44 | 2.20 | 2.03 | 1.89 | 2.47 | |
| Hydra-MDP (arXiv’24) [ 30 ] | 4.83 | 4.75 | 4.57 | 5.11 | 5.42 | 5.70 | 8.97 | 9.86 | 11.11 | 6.70 | |
| PARA-Drive (CVPR’24) [ 54 ] | 5.05 | 4.74 | 4.32 | 3.16 | 2.99 | 3.01 | 5.78 | 6.60 | 7.88 | 4.84 | |
| DiffusionDrive (CVPR’25) [ 33 ] | 4.47 | 4.47 | 4.27 | 4.05 | 3.95 | 3.91 | 3.54 | 3.43 | 3.28 | 3.93 |
| Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Mean |
| NC | 66.9 | 74.6 | 80.6 | 85.3 | 91.2 | 91.8 | 86.3 | 93.1 | 91.0 | 84.5 |
| DAC | 70.7 | 78.0 | 80.4 | 80.3 | 87.1 | 87.6 | 83.8 | 91.2 | 89.9 | 83.2 |
| TTC | 49.9 | 56.9 | 62.4 | 75.1 | 83.7 | 84.9 | 76.0 | 84.3 | 85.0 | 73.1 |
| Comf. | 45.8 | 55.9 | 69.6 | 79.3 | 80.9 | 82.3 | 84.0 | 76.6 | 63.0 | 70.8 |
| EP | 43.8 | 56.1 | 62.7 | 61.0 | 70.1 | 69.5 | 48.9 | 52.4 | 41.9 | 56.3 |
| PDMS | 35.2 | 47.0 | 54.4 | 59.6 | 71.1 | 71.7 | 56.7 | 65.1 | 58.3 | 57.7 |
| Metric | PCAT | PCMF | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. | |
| ADE | - | - | - | 4.47 | 4.47 | 4.27 | 4.05 | 3.95 | 3.91 | 3.54 | 3.43 | 3.28 | 3.93 |
| ✓ | - | - | 4.10 | 4.14 | 3.88 | 3.58 | 3.48 | 3.48 | 3.12 | 3.03 | 2.92 | 3.52 | |
| ✓ | ✓ | - | 3.80 | 3.82 | 3.55 | 3.29 | 3.20 | 3.18 | 2.85 | 2.77 | 2.63 | 3.23 | |
| - | ✓ | ✓ | 3.12 | 3.11 | 2.86 | 2.63 | 2.54 | 2.52 | 2.17 | 2.14 | 1.95 | 2.55 | |
| ✓ | - | ✓ | 3.22 | 3.21 | 2.96 | 2.69 | 2.59 | 2.55 | 2.19 | 2.12 | 1.99 | 2.61 | |
| ✓ | ✓ | ✓ | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 |
| Models | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| DeBERTa [ 17 ] | ADE | 9.27 | 9.10 | 8.60 | 6.22 | 5.86 | 6.52 | 2.19 | 3.13 | 3.02 | 5.99 |
| RoBERTa [ 37 ] | 3.98 | 4.49 | 4.12 | 4.74 | 4.08 | 2.92 | 4.53 | 6.20 | 7.43 | 4.72 | |
| MiniLM [ 51 ] | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| DeBERTa [ 17 ] | FDE | 20.78 | 19.52 | 17.61 | 10.76 | 9.97 | 11.56 | 3.32 | 5.39 | 5.11 | 11.56 |
| RoBERTa [ 37 ] | 8.35 | 9.54 | 8.24 | 8.47 | 6.95 | 4.97 | 7.85 | 10.91 | 13.58 | 8.76 | |
| MiniLM [ 51 ] | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Methods | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| DiffusionDrive (One-Hot) [ 33 ] | ADE | 3.32 | 3.29 | 3.03 | 2.71 | 2.65 | 2.59 | 2.24 | 2.16 | 2.03 | 2.67 |
| PersonaDrive (One-Hot) | 3.15 | 3.16 | 2.89 | 2.60 | 2.53 | 2.48 | 2.11 | 2.03 | 1.90 | 2.54 | |
| PersonaDrive (Text) | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| DiffusionDrive (One-Hot) [ 33 ] | FDE | 6.02 | 5.87 | 5.18 | 4.15 | 4.00 | 3.88 | 3.13 | 2.94 | 2.61 | 4.20 |
| PersonaDrive (One-Hot) | 5.66 | 5.60 | 4.87 | 3.98 | 3.86 | 3.76 | 3.01 | 2.83 | 2.52 | 4.01 | |
| PersonaDrive (Text) | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Methods | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| langTraj (ICCV’25) [ 4 ] | ADE | 3.64 | 3.35 | 3.17 | 2.93 | 2.46 | 2.47 | 2.37 | 2.09 | 2.01 | 2.72 |
| iMotionLLM (WACV’26) [ 12 ] | 3.44 | 3.51 | 3.33 | 2.97 | 2.94 | 2.88 | 2.37 | 2.29 | 2.15 | 2.87 | |
| PersonaDrive | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| langTraj (ICCV’25) [ 4 ] | FDE | 6.70 | 6.50 | 5.99 | 4.83 | 4.07 | 4.04 | 3.45 | 3.07 | 2.95 | 4.66 |
| iMotionLLM (WACV’26) [ 12 ] | 6.27 | 6.27 | 5.76 | 4.63 | 4.56 | 4.41 | 3.34 | 3.16 | 2.80 | 4.58 | |
| PersonaDrive | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Methods | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| 1D-Urg. | ADE | 3.44 | 3.05 | 3.00 | 2.98 | 2.47 | 2.40 | 2.79 | 1.94 | 2.37 | 2.72 |
| 1D-Comf. | 4.82 | 4.88 | 4.75 | 2.48 | 2.40 | 2.36 | 5.55 | 6.06 | 6.89 | 4.47 | |
| Multi-D | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| 1D-Urg. | FDE | 6.37 | 5.43 | 5.11 | 4.62 | 3.77 | 3.59 | 4.26 | 2.65 | 3.84 | 4.40 |
| 1D-Comf. | 11.91 | 11.53 | 10.61 | 3.75 | 3.60 | 3.51 | 9.48 | 10.48 | 12.37 | 8.58 | |
| Multi-D | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. | |||
| ADE | ✓ | - | - | 3.28 | 3.25 | 2.97 | 2.80 | 2.64 | 2.63 | 2.26 | 2.20 | 2.05 | 2.68 |
| ✓ | ✓ | - | 3.24 | 3.29 | 3.05 | 2.75 | 2.66 | 2.61 | 2.23 | 2.15 | 2.03 | 2.67 | |
| ✓ | - | ✓ | 3.02 | 3.05 | 2.79 | 2.49 | 2.41 | 2.40 | 2.05 | 2.01 | 1.88 | 2.46 | |
| ✓ | ✓ | ✓ | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| FDE | ✓ | - | - | 5.80 | 5.68 | 4.96 | 4.25 | 3.95 | 3.94 | 3.16 | 3.00 | 2.65 | 4.15 |
| ✓ | ✓ | - | 5.69 | 5.66 | 5.00 | 4.10 | 3.96 | 3.84 | 3.09 | 2.90 | 2.58 | 4.09 |
| Methods | Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. |
| Naive Concatenate | ADE | 3.79 | 3.79 | 3.69 | 3.46 | 3.68 | 3.38 | 3.08 | 2.94 | 2.77 | 3.40 |
| Naive Cross-Attention | 3.18 | 3.18 | 2.90 | 2.65 | 2.59 | 2.56 | 2.20 | 2.08 | 1.93 | 2.59 | |
| PCMF | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| Naive Concatenate | FDE | 6.44 | 6.33 | 5.94 | 5.17 | 5.49 | 4.94 | 4.30 | 4.05 | 3.60 | 5.14 |
| Naive Cross-Attention | 5.72 | 5.62 | 4.86 | 4.02 | 3.89 | 3.83 | 3.09 | 2.81 | 2.49 | 4.04 | |
| PCMF | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Metric | UH-CL | UH-CM | UH-CH | UM-CL | UM-CM | UM-CH | UL-CL | UL-CM | UL-CH | Avg. | ||
| ADE | ✓ | - | 3.27 | 3.25 | 3.02 | 2.73 | 2.64 | 2.59 | 2.26 | 2.20 | 2.05 | 2.67 |
| - | ✓ | 3.27 | 3.26 | 2.99 | 2.72 | 2.60 | 2.56 | 2.22 | 2.11 | 2.00 | 2.64 | |
| ✓ | ✓ | 2.98 | 2.96 | 2.69 | 2.43 | 2.35 | 2.34 | 2.02 | 1.93 | 1.82 | 2.39 | |
| FDE | ✓ | - | 5.75 | 5.63 | 4.95 | 4.10 | 3.92 | 3.83 | 3.12 | 2.95 | 2.64 | 4.10 |
| - | ✓ | 5.73 | 5.59 | 4.94 | 4.07 | 3.90 | 3.80 | 3.07 | 2.83 | 2.53 | 4.05 | |
| ✓ | ✓ | 5.25 | 5.15 | 4.47 | 3.69 | 3.52 | 3.50 | 2.82 | 2.63 | 2.33 | 3.71 |
| Judge | Top-1 | Urgency | Comfort | |
| Gemini Flash | 82.6 | 0.804 | 90.2 | 90.6 |
| Claude Opus | 98.0 | 0.978 | 100.0 | 98.0 |
| Configuration | Avg. ADE | Avg. FDE | Avg. PDMS |
| (a) KL temperature | |||
| 2.892 | 4.406 | 56.3 | |
| 2.391 | 3.706 | 57.7 | |
| 2.759 | 4.265 | 56.8 | |
| 2.772 | 4.274 | 57.4 | |
| (b) Inter-axis weight | |||
Multimodal Trajectory Prediction
Multimodal trajectory prediction estimates several plausible future paths for people, vehicles, and other agents from motion and scene context.
See all 56 papers