MoRAL: Compact VLM Sensor-Grounded BEV Inference for Edge Autonomous Driving

This paper introduces MoRAL, a compact Vision-Language Model (VLM) inference framework designed for resource-constrained autonomous driving platforms. By two-stage fine-tuning the Cosmos-Reason2-2B model, it can interpret physically encoded Bird's Eye View (BEV) representations to make driving decisions. The BEV image encodes LiDAR distance, object categories, and radar velocity into visual features, eliminating the need for 3D backbone networks during inference. The first stage fine-tunes the visual encoder on 60k grounded records; the second stage fine-tunes 2.4% of model parameters on 57k chain-of-thought records. On the nuScenes dataset, MoRAL outperforms the zero-shot 8B baseline in seven out of eight driving tasks, improving emergency braking recall from 10.8% to 47.8% and reducing output degradation from 94.1% to 20.8%, while running at 42 tok/s on an 8GB consumer GPU.

Background and Context

The deployment of Vision-Language Models (VLMs) in safety-critical autonomous driving applications has long been hindered by a fundamental contradiction between model size and spatial perception reliability. Traditional end-to-end autonomous driving systems typically rely on massive 3D backbone networks to interpret complex environmental data, a requirement that is computationally prohibitive for resource-constrained edge devices. These systems often struggle to balance the need for high-fidelity spatial understanding with the strict latency and memory limitations inherent in vehicle-mounted hardware. Consequently, there is a pressing industry need for architectures that can deliver robust decision-making capabilities without the overhead of heavy 3D reconstruction pipelines.

To address this challenge, researchers have introduced MoRAL, a compact inference framework designed specifically for edge autonomous driving platforms. MoRAL leverages the Cosmos-Reason2-2B model, a relatively lightweight VLM, but augments it with a novel "sensor-grounded" strategy. This approach fundamentally redefines how environmental data is presented to the model. Instead of processing raw sensor streams through complex neural networks, MoRAL encodes physical space information directly into visual features. This allows the compact VLM to "read" the driving environment as if it were interpreting a structured image, thereby bypassing the need for explicit 3D backbone networks during the inference phase. The framework aims to democratize advanced autonomous driving capabilities by making them accessible on consumer-grade hardware.

Deep Analysis

The technical core of MoRAL lies in its innovative representation of Bird's Eye View (BEV) data and its two-stage fine-tuning pipeline. The BEV image is engineered as a physically encoded representation where LiDAR distance is mapped to specific color bands, object categories are encoded through clustered morphologies, and radar Doppler velocity is visualized as directional wedge overlays. This transformation converts complex 3D spatial data into 2D visual features that a standard VLM can process natively. By externalizing spatial perception into the input image itself, the model eliminates the computational burden of 3D feature extraction during runtime, allowing the VLM to focus entirely on reasoning and decision-making based on these enriched visual cues.

The training process is rigorously divided into two distinct stages to ensure both visual comprehension and logical reasoning capabilities. In the first stage, the visual encoder is fine-tuned on 60,000 grounded records. This phase is critical for teaching the model to interpret the specific visual language of the encoded BEV images. Ablation studies confirm that without this explicit training, zero-shot baselines fail to parse the BEV outputs, indicating that the model must be explicitly taught to associate these visual patterns with physical entities. This stage effectively grounds the model's visual understanding in the physical reality of the driving environment.

The second stage focuses on enhancing the model's reasoning capabilities using a teacher-student distillation approach. The Cosmos-Reason2-8B model serves as the teacher, generating 57,696 Chain-of-Thought (CoT) records that cover eight distinct types of driving questions. During this phase, only 2.4% of the Cosmos-Reason2-2B model's parameters (approximately 52 million) are fine-tuned. This parameter-efficient strategy allows the smaller model to inherit complex multi-step physical reasoning abilities from the larger teacher model while retaining its pre-trained language capabilities. The result is a model that can perform structured, logical deductions based on the visual input, rather than merely recognizing objects.

Industry Impact

Empirical evaluations conducted on the nuScenes dataset demonstrate the efficacy of the MoRAL framework. Using a subset of 2,304 frames, the model was assessed by a Gemma 4 (31B) evaluator and calibrated through human review. MoRAL outperformed the zero-shot 8B baseline in seven out of eight driving task categories, despite having only one-quarter of the parameters. The most significant improvements were observed in tasks requiring structured multi-step physical reasoning. Specifically, the recall rate for emergency braking scenarios jumped from 10.8% in the baseline to 47.8% in MoRAL. Furthermore, the rate of output degradation, which indicates incoherent or irrelevant responses, dropped dramatically from 94.1% to 20.8%, signaling a substantial increase in the reliability and coherence of the model's decisions.

Beyond accuracy, MoRAL offers compelling efficiency advantages for edge deployment. The entire inference pipeline operates without the need for quantization, running at 42 tokens per second on a consumer-grade GPU with just 8GB of VRAM. This performance metric is particularly significant for the automotive industry, as it proves that high-fidelity autonomous driving decisions can be made on affordable, widely available hardware. This capability lowers the barrier to entry for developing advanced driver-assistance systems (ADAS), potentially accelerating the adoption of L2+/L3 autonomous features in mass-market vehicles. The ability to achieve real-time inference on such modest hardware challenges the prevailing trend of relying on expensive, high-end computing units for every vehicle.

The implications for the open-source autonomous driving community and industrial R&D are profound. MoRAL demonstrates that sophisticated physical reasoning does not strictly require massive model scaling. By focusing on clever data representation and parameter-efficient fine-tuning, researchers can unlock significant performance gains in compact models. This paradigm shift encourages the community to explore more efficient ways of integrating physical priors into VLMs. It suggests that future advancements may come not from simply increasing parameter counts, but from better structuring input data to align with the innate capabilities of smaller, more agile models.

Outlook

The success of MoRAL points toward a future where autonomous driving systems are more modular, efficient, and accessible. By decoupling spatial perception from the reasoning engine, this architecture offers a flexible foundation for future iterations. As sensor technologies evolve, the BEV encoding layer can be adapted to incorporate new data types without requiring a complete redesign of the VLM backbone. This modularity could lead to faster development cycles and easier integration of new sensing modalities into existing autonomous stacks.

Moreover, the emphasis on edge efficiency aligns with the growing industry focus on privacy and latency. Processing data locally on the vehicle reduces the dependency on cloud connectivity, which is crucial for real-time safety applications. As computational constraints remain a primary bottleneck for widespread autonomous adoption, frameworks like MoRAL provide a viable pathway to overcome these limitations. The industry may see a shift towards hybrid architectures that combine lightweight, sensor-grounded VLMs for immediate decision-making with larger, cloud-based models for long-horizon planning and learning.

Finally, the rigorous evaluation methodology employed in this study sets a new standard for benchmarking compact autonomous driving models. The use of automated evaluators calibrated by human review ensures that performance metrics reflect true driving safety rather than superficial pattern matching. As the field matures, such rigorous standards will be essential for validating the safety of increasingly autonomous systems. MoRAL serves as a testament to the potential of combining domain-specific data engineering with modern large language model techniques to solve complex engineering problems efficiently.

Sources