Predictive processing as a scalable computational principle for embodied multitask intelligence
Data files
Jul 07, 2026 version files 190.66 MB
-
README.md
9.99 KB
-
robot_data_learning_test.zip
190.65 MB
Abstract
Humans exhibit remarkable flexibility in adapting to diverse and uncertain environments—a hallmark arising from the brain’s ability to integrate multimodal sensory streams into coherent predictive models. Drawing on this principle, we introduce a scalable hierarchical multimodal recurrent neural network grounded in predictive processing under the free-energy principle, capable of directly integrating over 30,000-dimensional visuo-proprioceptive inputs without dimensionality reduction or handcrafted preprocessing. Using sensory data from teleoperation of a full-scale physical humanoid robot performing two caregiving-related tasks—rigid-body repositioning and flexible-towel wiping—the model learns to predict high-dimensional visuo-proprioceptive streams end-to-end. In open-loop adaptive inference experiments, the framework exhibits three emergent properties: (i) self-organized hierarchical latent dynamics governing task transitions, uncertainty, and occlusion inference; (ii) robustness to degraded vision via multimodal integration; and (iii) asymmetric interference in multitask learning. Although evaluated in simulations, the framework is extensible to closed-loop robot control, with proprioceptive predictions driving action, thereby establishing a generalizable computational foundation bridging brain theory, artificial intelligence, and embodied robotics.
This dataset contains the visuo-proprioceptive time-series data used to train and test a scalable predictive-coding-inspired variational recurrent neural network (scalable PV-RNN) for embodied multitask learning. The data were recorded while a Dry-AIREC dual-arm humanoid robot was teleoperated to perform two caregiving-related tasks with a mannequin on a bed: repositioning the mannequin from a supine posture toward sitting, and wiping the mannequin with a towel. The dataset includes the visual image streams and proprioceptive streams used as observed inputs and target sequences for neural network learning and test inference.
Description of the data and file structure
The submitted data file is robot_data_learning_test.zip. When uncompressed, it contains one top-level folder:
robot_data_learning_test/
learning_data/
proprio_0000000.txt ... proprio_0000017.txt
img_16163_c1_0000000/ ... img_64643_c2_0000017/
test_data/
proprio_0000000.txt ... proprio_0000005.txt
img_16163_c1_0000000/ ... img_64643_c2_0000005/
learning_data/ contains the 18 teleoperated sequences used for model learning. test_data/ contains 6 held-out teleoperated sequences used for simulation-based online inference tests after learning. Each sequence consists of one proprioceptive text file and six corresponding image folders: three image resolutions for each of two head-mounted RGB cameras.
The robot sensory streams were recorded at 20 Hz. Each row in a proprioceptive text file and each PNG image frame corresponds to the same time step. For example, in trial 0000000, row 1 of proprio_0000000.txt corresponds to image frame 0000000.png in each img_*_0000000/ folder, row 2 corresponds to 0000001.png, and so on.
Experimental context and trial groups
The study asked whether one predictive-processing neural network can learn and generalize across heterogeneous caregiving tasks from high-dimensional multimodal sensory streams without handcrafted feature extraction or dimensionality reduction.
The two tasks are:
- Repositioning: a rigid-body manipulation task in which the robot reaches toward the mannequin, supports the upper back and legs, and lifts the upper body. This task contains the subtask progression reaching, holding, and lifting.
- Wiping: a flexible-object surface-interaction task in which the robot reaches for a towel placed on the mannequin, wipes the body surface, and completes/releases the wiping motion. This task contains the subtask progression reaching, wiping, and releasing.
Both tasks were performed under variability in initial mannequin posture, action timing, and bed height. The bed heights used in the experiment were 590, 650, and 710 mm. Bed height is part of the experimental condition represented by the recorded sequences but is not encoded directly in the file or folder names.
Trial IDs are zero-padded and identify corresponding proprioceptive and image data within each split:
| Split | Trial IDs | Task | Experimental role |
|---|---|---|---|
learning_data |
0000000-0000008 |
Repositioning | 9 learning trials: 3 bed heights x 3 repetitions |
learning_data |
0000009-0000017 |
Wiping | 9 learning trials: 3 bed heights x 3 repetitions |
test_data |
0000000-0000002 |
Repositioning | 3 held-out test sequences: 1 per bed height |
test_data |
0000003-0000005 |
Wiping | 3 held-out test sequences: 1 per bed height |
During learning, the model was trained to reconstruct and predict the observed visuo-proprioceptive sequences. During testing, trained models used the six held-out sequences for online inference and generated one-step-ahead predictions of the visual and proprioceptive streams at each time step. Those generated prediction outputs are not included in this Dryad data package; the deposited files are the recorded observed sequences used as learning and test data.
Proprioceptive data files
Files named proprio_<trial_id>.txt contain the robot proprioceptive stream for one trial.
- Format: plain text, tab-delimited numeric matrix, no header row.
- Rows: time steps sampled at 20 Hz.
- Columns: 28 normalized proprioceptive variables.
- Column contents: columns 1-14 are joint-angle values for the two 7-DOF arms; columns 15-28 are the corresponding joint-torque values.
- Normalization: values were normalized to the range -0.9 to 0.9 using maximum and minimum values, matching the activation range used by the neural network.
The number of rows differs by trial because task execution duration varied. The row count in each proprioceptive file matches the number of PNG frames in each corresponding image folder for that trial.
Visual image data folders
Folders named img_<resolution_code>_<camera_id>_<trial_id>/ contain the visual stream for one trial, image resolution, and camera.
The naming convention is:
img: visual image data.<resolution_code>: RGB image size and channels.16163= 16 x 16 RGB,32323= 32 x 32 RGB, and64643= 64 x 64 RGB.<camera_id>: camera identifier.c1andc2are the two binocular/head-mounted RGB camera views.<trial_id>: zero-padded seven-digit trial ID matching the associatedproprio_<trial_id>.txtfile.
Each image folder contains sequential PNG frames named 0000000.png, 0000001.png, etc. The frame number is the time-step index within that trial. PNG files are 8-bit RGB images.
For each trial, six image folders are provided:
img_16163_c1_<trial_id>/img_16163_c2_<trial_id>/img_32323_c1_<trial_id>/img_32323_c2_<trial_id>/img_64643_c1_<trial_id>/img_64643_c2_<trial_id>/
The neural network input combined binocular RGB images at all three resolutions. Across both cameras and all resolutions, the visual input dimensionality per time step was 32,256 values: 2 cameras x 3 color channels x (16 x 16 + 32 x 32 + 64 x 64 pixels).
File counts and sequence lengths
The uncompressed dataset contains 24 proprioceptive text files and 45,546 PNG image files.
Learning data:
| Trial ID | Task | Time steps / proprioceptive rows | PNG frames per image folder |
|---|---|---|---|
0000000 |
Repositioning | 350 | 350 |
0000001 |
Repositioning | 335 | 335 |
0000002 |
Repositioning | 301 | 301 |
0000003 |
Repositioning | 518 | 518 |
0000004 |
Repositioning | 323 | 323 |
0000005 |
Repositioning | 357 | 357 |
0000006 |
Repositioning | 462 | 462 |
0000007 |
Repositioning | 420 | 420 |
0000008 |
Repositioning | 335 | 335 |
0000009 |
Wiping | 300 | 300 |
0000010 |
Wiping | 295 | 295 |
0000011 |
Wiping | 240 | 240 |
0000012 |
Wiping | 250 | 250 |
0000013 |
Wiping | 230 | 230 |
0000014 |
Wiping | 230 | 230 |
0000015 |
Wiping | 270 | 270 |
0000016 |
Wiping | 270 | 270 |
0000017 |
Wiping | 230 | 230 |
Test data:
| Trial ID | Task | Time steps / proprioceptive rows | PNG frames per image folder |
|---|---|---|---|
0000000 |
Repositioning | 410 | 410 |
0000001 |
Repositioning | 320 | 320 |
0000002 |
Repositioning | 455 | 455 |
0000003 |
Wiping | 230 | 230 |
0000004 |
Wiping | 230 | 230 |
0000005 |
Wiping | 230 | 230 |
Sharing/Access information
Links to other publicly accessible locations of the data:
- Dryad dataset DOI: https://doi.org/10.5061/dryad.qv9s4mwvs
Data were derived from:
- Original teleoperation recordings of the Dry-AIREC humanoid robot collected for the related study. The deposited files are the processed and normalized visual and proprioceptive sequences used for model learning and test evaluation.
Code/Software
The neural network code associated with this dataset is available at:
The data files can be read with standard tools for PNG images and tab-delimited text. No specialized software is required simply to inspect the deposited data. To reproduce the model learning and testing workflow, use the associated scalable PV-RNN code repository.
The data was collected with a humanoid robot (called Dry-AIREC) equipped with proprioceptive and visual sensors.
