FLEET: Token-Based Feature Extraction for Event Camera-based Reinforcement Learning
Organizations: Institute of Information Processing, L3S, Leibniz University Hannover, Germany
Abstract
Event cameras generate asynchronous, high-frequency data streams offering spatially sparse information at lower latency than traditional cameras. In principle, these properties should be ideal for the design of control policies. However, reinforcement learning research in this field remains limited as existing approaches fail to fully exploit the sensor's properties. CNN-based methods negate the sensors benefits by aggregating events into sparse grids. This couples compute cost to sensor resolution and blurs the temporal information. Meanwhile, existing generative baselines rely on the availability of trajectory data to pretrain the model. We propose FLEET (Feature Learning from Events via Efficient Tokenization), a feature extractor that processes event sequences directly. Leveraging random Fourier features and cross-attention, our architecture compresses variable streams into fixed-size latent representations. This decouples inference cost of the feature extractor's backbone from the sensor's resolution, enabling end-to-end learning without auxiliary losses. We validate FLEET on a new, high-throughput benchmark. The results demonstrate that our sequence-based approach surpasses SOTA performance and exhibits superior robustness to variations in observation frequencies.
Figures & tables
| Feature Extractor | Racing (Return ) | Parking (Return ) | Inv. Dbl. Pendulum (Return ) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| IQM | CI95 | IQM | CI95 | IQM | CI95 | ||||
| NatureCNN (RGB) [ 34 ] | 23.89 | [12.71, | 35.89] | 5.01 | [4.50, | 5.41] | 8980.69 | [7883.73, | 9352.29] |
| NatureCNN [ 34 ] | 6.83 | [4.41, | 11.44] | 3.89 | [3.50, | 4.82] | 560.50 | [469.09, | 1013.37] |
| ImpalaCNN [ 12 ] | 4.81 | [4.39, | 5.93] | 5.18 | [4.77, | 5.59] | 1175.57 | [184.60, | 2178.99] |
| LFF-DS [ 43 ] | 5.36 | [4.22, | 6.46] | 0.40 | [0.01, | 0.81] | 848.29 | [788.83, | 919.08] |
| eVAE [ 53 ] | 1.13 | [0.81, | 1.43] | 0.82 | [0.78, | 0.90] | 429.67 | [414.25, | 441.36] |
| Feature Extractor | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| (Return ) | (Return ) | (Return ) | (Return ) | |||||||||
| IQM | CI95 | IQM | CI95 | IQM | CI95 | IQM | CI95 | |||||
| NatureCNN | 6.83 | [4.41, | 11.44] | 11.43 | [9.18, | 14.70] | 16.42 | [11.88, | 21.25] | 13.21 | [10.64, | 17.55] |
| DMRCNN | 12.94 | [9.09, | 16.99] | 9.55 | [6.90, | 13.02] | 10.46 | [8.08, | 13.88] | 8.57 | [6.41 , | 12.62] |
| FLEET ( ours ) | 46.96 | [43.28, | 49.03] | 40.48 | [36.49, | 44.17] | 43.65 | [36.85, | 49.64] | 44.32 | [39.89, | 49.57] |
| Feature Extractor | |||||||||
|---|---|---|---|---|---|---|---|---|---|
| (Return ) | (Return ) | (Return ) | |||||||
| IQM | CI95 | IQM | CI95 | IQM | CI95 | ||||
| NatureCNN [ 34 ] | 19.02 | [16.82, | 21.08] | 6.83 | [4.41, | 11.44] | 5.87 | [4.10, | 10.40] |
| DMRCNN [ 55 ] | 21.00 | [19.70, | 22.62] | 12.94 | [9.09, | 16.99] | 6.78 | [5.36, | 7.71] |
| FLEET ( ours ) | 40.33 | [38.43, | 42.10] | 46.96 | [43.28, | 49.03] | 41.92 | [34.18, | 46.68] |
| Hyperparameter | Value |
|---|---|
| FLEET | |
| Event Sequence Length ( ) | 2048 |
| Embedding Dimension ( ) | 128 |
| 5.0 | |
| 20.0 | |
| Number of Latent Tokens ( ) | 32 |
| Hyperparameter | Value |
|---|---|
| NatureCNN | |
| Conv. Architecture (Channels, Kernel, Stride) | |
| Output Dimension | 512 |
| ImpalaCNN | |
| Conv. Arch. (Features, Kernel, Stride, ResBlocks) | |
| Output Dimension | 256 |
| Feature Extractor | ||||||||
|---|---|---|---|---|---|---|---|---|
| GFLOPs( ) | VRAM GB | GFLOPs( ) | VRAM GB | GFLOPs( ) | VRAM GB | GFLOPs( ) | VRAM GB | |
| NatureCNN | 175.11 | 1.25 | 353.93 | 2.49 | 712.84 | 4.98 | 2876.33 | 19.92 |
| DMRCNN | 47.11 | 0.54 | 108.05 | 1.12 | 236.07 | 2.28 | 1053.09 | 9.37 |
| FLEET ( ours ) | 83.28 | 4.09 | 83.28 | 4.09 | 83.28 | 4.09 | 83.28 | 4.09 |
| Feature Extractor | ||||
|---|---|---|---|---|
| Exec. Time | Exec. Time | Exec. Time | Exec. Time | |
| NatureCNN | 286.63 | 577.67 | 1165.83 | 4778.59 |
| DMRCNN | 81.48 | 260.72 | 644.07 | 3690.64 |
| FLEET ( ours ) | 11.76 | 11.76 | 11.76 | 11.76 |
| Sequence Sampling Technique | Return | ||
|---|---|---|---|
| IQM | CI95 | Rel. Change | |
| FLEET ( ours ) | 46.96 | [43.28, 49.03] | - |
| Temporal Sampling | 37.02 | [34.68, 39.89] | -21% |
| Sequence Sampling Technique | Return | ||
|---|---|---|---|
| IQM | CI95 | Rel. Change | |
| FLEET ( ours ) | 46.96 | [43.28, 49.03] | - |
| Matched Sequence Baseline | 30.06 | [26.13, 36.12] | -35% |
| Feature Extractor | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| (Return ) | (Return ) | (Return ) | (Return ) | (Return ) | |||||||||||
| IQM | CI95 | IQM | CI95 | IQM | CI95 | IQM | CI95 | IQM | CI95 | ||||||
| FLEET ( ours ) | 38.81 | [35.99, | 41.09] | 41.55 | [38.94, | 44.57] | 46.96 | [43.28, | 49.03] | 43.56 | [39.51, | 47.71] | 41.59 | [37.11, | 48.52] |
| Feature Extractor | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| (Return ) | (Return ) | (Return ) | (Return ) | |||||||||
| IQM | CI95 | IQM | CI95 | IQM | CI95 | IQM | CI95 | |||||
| FLEET ( ours ) | 40.05 | [37.23, | 44.14] | 41.39 | [36.00, | 43.41] | 46.96 | [43.28, | 49.03] | 44.04 | [40.80, | 47.23] |
| Aggregation Technique | Return | ||
|---|---|---|---|
| IQM | CI95 | Rel. Change | |
| Temporal Gating | 46.96 | [43.28, 49.03] | - |
| Addition | 40.03 | [36.47, 44.25] | -14% |
| Multiplication | 31.83 | [27.39, 35.80] | -32% |
| Concatenation | 39.74 | [34.35, 45.91] | -15% |
Representation Learning for RL
Representation learning for reinforcement learning builds state, task, and dynamics features that help agents learn, explore, and transfer.
See all 69 papers