元記事の説明文
<!-- twitter -->
Training Diffusion Models with Reinforcement Learning
<video loop="" style="width: 100%; margin: 0; padding: 0; border: none; background: transparent; display: block; border-radius: 5px;">
<source src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/megavandertest.mp4" type="video/mp4" />
</video>
<p style="margin-top: 20px;">
<b>We deployed 100 reinforcement learning (RL)-controlled cars into rush-hour highway traffic to smooth congestion and reduce fuel consumption for everyone.</b> Our goal is to tackle <a href="https://www.youtube.com/watch?v=TNokBgtSUvQ" target="_blank">"stop-and-go" waves</a>, those frustrating slowdowns and speedups that usually have no clear cause but lead to congestion and significant energy waste. To train efficient flow-smoothing controllers, we built fast, data-driven simulations that RL agents interact with, learning to maximize energy efficiency while maintaining throughput and operating safely around human drivers.
</p>
<p>
Overall, a small proportion of well-controlled autonomous vehicles (AVs) is enough to significantly improve traffic flow and fuel efficiency for all drivers on the road. Moreover, the trained controllers are designed to be deployable on most modern vehicles, operating in a decentralized manner and relying on standard radar sensors. In our <a href="https://ieeexplore.ieee.org/document/10858625" target="_blank">latest paper</a>, we explore the challenges of deploying RL controllers on a large-scale, from simulation to the field, during this 100-car experiment.
</p>
<!--more-->
<h2 id="the-challenges-of-phantom-jams">The challenges of phantom jams</h2>
<p style="text-align: center; margin-top: 50px;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/highway_wave.gif" style="width: 80%; border-radius: 5px;" width="80%" />
<br />
<i>A stop-and-go wave moving backwards through highway traffic.</i>
</p>
<p>If you drive, you’ve surely experienced the frustration of stop-and-go waves, those seemingly inexplicable traffic slowdowns that appear out of nowhere and then suddenly clear up. These waves are often caused by small fluctuations in our driving behavior that get amplified through the flow of traffic. We naturally adjust our speed based on the vehicle in front of us. If the gap opens, we speed up to keep up. If they brake, we also slow down. But due to our nonzero reaction time, we might brake just a bit harder than the vehicle in front. The next driver behind us does the same, and this keeps amplifying. Over time, what started as an insignificant slowdown turns into a full stop further back in traffic. These waves move backward through the traffic stream, leading to significant drops in energy efficiency due to frequent accelerations, accompanied by increased CO<sub>2</sub> emissions and accident risk.</p>
<p>And this isn’t an isolated phenomenon! These waves are ubiquitous on busy roads when the traffic density exceeds a critical threshold. So how can we address this problem? Traditional approaches like ramp metering and variable speed limits attempt to manage traffic flow, but they often require costly infrastructure and centralized coordination. A more scalable approach is to use AVs, which can dynamically adjust their driving behavior in real-time. However, simply inserting AVs among human drivers isn’t enough: they must also drive in a smarter way that makes traffic better for everyone, which is where RL comes in.</p>
<p style="text-align: justify; margin-top: 50px;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/fundamental_diagram.png" style="display: block; margin: auto;" width="80%" />
<br />
<i><b>Fundamental diagram of traffic flow.</b> The number of cars on the road (density) affects how much traffic is moving forward (flow). At low density, adding more cars increases flow because more vehicles can pass through. But beyond a critical threshold, cars start blocking each other, leading to congestion, where adding more cars actually slows down overall movement.</i>
</p>
<h2 id="reinforcement-learning-for-wave-smoothing-avs">Reinforcement learning for wave-smoothing AVs</h2>
<p>RL is a powerful control approach where an agent learns to maximize a reward signal through interactions with an environment. The agent collects experience through trial and error, learns from its mistakes, and improves over time. In our case, the environment is a mixed-autonomy traffic scenario, where AVs learn driving strategies to dampen stop-and-go waves and reduce fuel consumption for both themselves and nearby human-driven vehicles.</p>
<p>Training these RL agents requires fast simulations with realistic traffic dynamics that can replicate highway stop-and-go behavior. To achieve this, we leveraged experimental data collected on Interstate 24 (I-24) near Nashville, Tennessee, and used it to build simulations where vehicles replay highway trajectories, creating unstable traffic that AVs driving behind them learn to smooth out.</p>
<p style="text-align: center; margin-top: 50px;">
<video loop="" style="width: 100%; margin: 0; padding: 0; border: none; background: transparent; display: block; border-radius: 5px;">
<source src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/simulation.mp4" type="video/mp4" />
</video>
<br />
<i>Simulation replaying a highway trajectory that exhibits several stop-and-go waves.</i>
</p>
<p>We designed the AVs with deployment in mind, ensuring that they can operate using only basic sensor information about themselves and the vehicle in front. The observations consist of the AV’s speed, the speed of the leading vehicle, and the space gap between them. Given these inputs, the RL agent then prescribes either an instantaneous acceleration or a desired speed for the AV. The key advantage of using only these local measurements is that the RL controllers can be deployed on most modern vehicles in a decentralized way, without requiring additional infrastructure.</p>
<h3 id="reward-design">Reward design</h3>
<p>The most challenging part is designing a reward function that, when maximized, aligns with the different objectives that we desire the AVs to achieve:</p>
<ul>
<li><strong>Wave smoothing:</strong> Reduce stop-and-go oscillations.</li>
<li><strong>Energy efficiency:</strong> Lower fuel consumption for all vehicles, not just AVs.</li>
<li><strong>Safety:</strong> Ensure reasonable following distances and avoid abrupt braking.</li>
<li><strong>Driving comfort:</strong> Avoid aggressive accelerations and decelerations.</li>
<li><strong>Adherence to human driving norms:</strong> Ensure a “normal” driving behavior that doesn’t make surrounding drivers uncomfortable.</li>
</ul>
<p>Balancing these objectives together is difficult, as suitable coefficients for each term must be found. For instance, if minimizing fuel consumption dominates the reward, RL AVs learn to come to a stop in the middle of the highway because that is energy optimal. To prevent this, we introduced dynamic minimum and maximum gap thresholds to ensure safe and reasonable behavior while optimizing fuel efficiency. We also penalized the fuel consumption of human-driven vehicles behind the AV to discourage it from learning a selfish behavior that optimizes energy savings for the AV at the expense of surrounding traffic. Overall, we aim to strike a balance between energy savings and having a reasonable and safe driving behavior.</p>
<h3 id="simulation-results">Simulation results</h3>
<p style="text-align: center; margin-top: 0;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/gap_thresholds.png" width="80%" />
<br />
<i>Illustration of the dynamic minimum and maximum gap thresholds, within which the AV can operate freely to smooth traffic as efficiently as possible.</i>
</p>
<p>The typical behavior learned by the AVs is to maintain slightly larger gaps than human drivers, allowing them to absorb upcoming, possibly abrupt, traffic slowdowns more effectively. In simulation, this approach resulted in significant fuel savings of up to 20% across all road users in the most congested scenarios, with fewer than 5% of AVs on the road. And these AVs don’t have to be special vehicles! They can simply be standard consumer cars equipped with a smart adaptive cruise control (ACC), which is what we tested at scale.</p>
<p style="text-align: justify; margin-top: 50px;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/wave_smoothing.png" style="display: block; margin: auto;" width="100%" />
<i>
<b>Smoothing behavior of RL AVs.</b> Red: a human trajectory from the dataset. Blue: successive AVs in the platoon, where AV 1 is the closest behind the human trajectory. There is typically between 20 and 25 human vehicles between AVs. Each AV doesn’t slow down as much or accelerate as fast as its leader, leading to decreasing wave amplitude over time and thus energy savings.
</i>
</p>
<h2 id="100-av-field-test-deploying-rl-at-scale">100 AV field test: deploying RL at scale</h2>
<div style="display: flex; width: 100%; margin-top: 30px;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/parking_lot.png" style="height: 300px; width: 50%; border-top-left-radius: 5px; border-bottom-left-radius: 5px;" />
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/parking_lot_drone.png" style="height: 300px; width: 50%; border-top-right-radius: 5px; border-bottom-right-radius: 5px;" />
</div>
<p style="text-align: center; margin-top: 10px;">
<i style="font-size: 0.9rem;">Our 100 cars parked at our operational center during the experiment week.</i>
</p>
<p>Given the promising simulation results, the natural next step was to bridge the gap from simulation to the highway. We took the trained RL controllers and deployed them on 100 vehicles on the I-24 during peak traffic hours over several days. This large-scale experiment, which we called the MegaVanderTest, is the largest mixed-autonomy traffic-smoothing experiment ever conducted.</p>
<p>Before deploying RL controllers in the field, we trained and evaluated them extensively in simulation and validated them on the hardware. Overall, the steps towards deployment involved:</p>
<ul>
<li><strong>Training in data-driven simulations:</strong> We used highway traffic data from I-24 to create a training environment with realistic wave dynamics, then validate the trained agent’s performance and robustness in a variety of new traffic scenarios.</li>
<li><strong>Deployment on hardware:</strong> After being validated in robotics software, the trained controller is uploaded onto the car and is able to control the set speed of the vehicle. We operate through the vehicle’s on-board cruise control, which acts as a lower-level safety controller.</li>
<li><strong>Modular control framework:</strong> One key challenge during the test was not having access to the leading vehicle information sensors. To overcome this, the RL controller was integrated into a hierarchical system, the MegaController, which combines a speed planner guide that accounts for downstream traffic conditions, with the RL controller as the final decision maker.</li>
<li><strong>Validation on hardware:</strong> The RL agents were designed to operate in an environment where most vehicles were human-driven, requiring robust policies that adapt to unpredictable behavior. We verify this by driving the RL-controlled vehicles on the road under careful human supervision, making changes to the control based on feedback.</li>
</ul>
<div style="display: flex; width: 100%; margin: 30px 0;">
<div style="display: flex; width: 48%;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/raspberry_pi.png" style="height: 200px; width: 100%; border-radius: 5px;" />
<i style="font-size: 0.9rem; display: block; text-align: center; margin-top: 5px;">Each of the 100 cars is connected to a Raspberry Pi, on which the RL controller (a small neural network) is deployed.</i>
</div>
<div style="display: flex; width: 48%;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/acc.png" style="height: 200px; width: 100%; border-radius: 5px;" />
<i style="font-size: 0.9rem; display: block; text-align: center; margin-top: 5px;">The RL controller directly controls the onboard adaptive cruise control (ACC) system, setting its speed and desired following distance.</i>
</div>
</div>
<p>Once validated, the RL controllers were deployed on 100 cars and driven on I-24 during morning rush hour. Surrounding traffic was unaware of the experiment, ensuring unbiased driver behavior. Data was collected during the experiment from dozens of overhead cameras placed along the highway, which led to the extraction of millions of individual vehicle trajectories through a computer vision pipeline. Metrics computed on these trajectories indicate a trend of reduced fuel consumption around AVs, as expected from simulation results and previous smaller validation deployments. For instance, we can observe that the closer people are driving behind our AVs, the less fuel they appear to consume on average (which is calculated using a calibrated energy model):</p>
<p style="text-align: center; margin-top: 0;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/fuel_data.png" width="80%" />
<br />
<i>Average fuel consumption as a function of distance behind the nearest engaged RL-controlled AV in the downstream traffic. As human drivers get further away behind AVs, their average fuel consumption increases.</i>
</p>
<p>Another way to measure the impact is to measure the variance of the speeds and accelerations: the lower the variance, the less amplitude the waves should have, which is what we observe from the field test data. Overall, although getting precise measurements from a large amount of camera video data is complicated, we observe a trend of 15 to 20% of energy savings around our controlled cars.</p>
<p style="text-align: center; margin-top: 0;">
<img src="https://bair.berkeley.edu/static/blog/rl_av_smoothing/data_scatter.png" width="50%" />
<br />
<i>Data points from all vehicles on the highway over a single day of the experiment, plotted in speed-acceleration space. The cluster to the left of the red line represents congestion, while the one on the right corresponds to free flow. We observe that the congestion cluster is smaller when AVs are present, as measured by computing the area of a soft convex envelope or by fitting a Gaussian kernel.</i>
</p>
<h2 id="final-thoughts">Final thoughts</h2>
<p>The 100-car field operational test was decentralized, with no explicit cooperation or communication between AVs, reflective of current autonomy deployment, and bringing us one step closer to smoother, more energy-efficient highways. Yet, there is still vast potential for improvement. Scaling up simulations to be faster and more accurate with better human-driving models is crucial for bridging the simulation-to-reality gap. Equipping AVs with additional traffic data, whether through advanced sensors or centralized planning, could further improve the performance of the controllers. For instance, while multi-agent RL is promising for improving cooperative control strategies, it remains an open question how enabling explicit communication between AVs over 5G networks could further improve stability and further mitigate stop-and-go waves. Crucially, our controllers integrate seamlessly with existing adaptive cruise control (ACC) systems, making field deployment feasible at scale. The more vehicles equipped with smart traffic-smoothing control, the fewer waves we’ll see on our roads, meaning less pollution and fuel savings for everyone!</p>
<hr />
<p><i>Many contributors took part in making the MegaVanderTest happen! The full list is available on the <a href="https://circles-consortium.github.io/" target="_blank">CIRCLES project</a> page, along with more details about the project.</i></p>
<p><i><b>Read more: <a href="https://ieeexplore.ieee.org/document/10858625" target="_blank">[paper]</a></b></i></p>