Skip to content

Neural Networks for Robotics

Conceptual

Neural Networks are computational models inspired by biological neurons that learn patterns from data. In robotics, they power perception (object detection, depth estimation), decision-making (policy learning), and increasingly, end-to-end control.

Traditional robotics relied on hand-crafted algorithms. Neural networks excel when:

  • Input is high-dimensional — cameras, LiDAR produce millions of data points
  • Patterns are complex — object recognition, natural language
  • Rules are hard to specify — dexterous manipulation, social navigation
  • Adaptation is needed — handle novel situations not seen during programming

Process spatial data like images. Used for:

  • Object detection (YOLO, Faster R-CNN)
  • Semantic segmentation (U-Net, DeepLab)
  • Depth estimation (MiDaS, DepthAnything)
Image → [Conv] → [Conv] → [Pool] → [FC] → Detection
↓ ↓
Features Features

Attention-based architecture. Originally for language, now dominant in:

  • Vision (ViT, DINO)
  • Multi-modal models (VLMs)
  • Robot foundation models (RT-2, GR00T)
Input Tokens → [Self-Attention] → [FFN] → Output
↓
"What's relevant?"

Process sequences. Used for:

  • Trajectory prediction
  • Language understanding
  • Time-series sensor data

Increasingly replaced by Transformers.

┌─────────────────────────────────────────────────┐
│ Perception │
│ CNN: Object detection, segmentation │
│ ViT: Scene understanding │
│ Depth networks: 3D reconstruction │
├─────────────────────────────────────────────────┤
│ Understanding │
│ VLM: "What do I see?" + "What should I do?" │
│ LLM: Task planning from language │
├─────────────────────────────────────────────────┤
│ Policy │
│ Imitation learning: Learn from demonstrations │
│ RL: Learn from trial and error │
│ VLA: Vision-Language-Action models │
├─────────────────────────────────────────────────┤
│ Control │
│ Neural network controllers │
│ Residual learning on top of classical control │
└─────────────────────────────────────────────────┘

Learn from labeled examples: (input, correct_output) pairs

Robotics use: Object detection, segmentation, depth estimation

Learn from rewards: trial and error

Robotics use: Locomotion, manipulation, game playing

Learn from demonstrations: watch expert, mimic behavior

Robotics use: Manipulation tasks, driving

Learn from data structure itself: no labels needed

Robotics use: Pre-training visual representations (DINO, MAE)

Real-time robotics requires fast inference:

HardwareTypical Inference
CPU only100ms+
GPU (desktop)5-20ms
Jetson Orin10-30ms
Jetson Thor5-15ms
TensorRT optimized2-10ms

TensorRT (NVIDIA) — Optimizes models for Jetson/GPU:

  • Layer fusion
  • Precision reduction (FP32 → FP16 → INT8)
  • Kernel auto-tuning
Terminal window
# Convert PyTorch model to TensorRT
trtexec --onnx=model.onnx --saveEngine=model.trt --fp16
  • Edge (Jetson): Low latency, works offline, limited compute
  • Cloud: Unlimited compute, high latency, requires connectivity

Most robots use edge inference for real-time control, cloud for heavy lifting.

Large models pre-trained on massive data, fine-tuned for robotics:

ModelTypeUse Case
RT-2VLAManipulation from language
GR00T N1.6VLAHumanoid full-body control
OpenVLAVLAGeneralizable manipulation (open-weight)
PaLM-EVLMEmbodied reasoning
OctoPolicyGeneralizable manipulation

These run efficiently on Jetson Thor with Transformer Engine.