This repository accompanies the research paper "Precision Angle Seeking in Robots: A Reinforcement Learning Approach in Simulation and Reality." Our study investigates the application of Reinforcement Learning (RL) and Deep Reinforcement Learning (DRL) algorithms for precision angle seeking in robotic control, highlighting the transition from simulations to real-world applications. The "Angular Positioning Seeker (APS)" environment, built on stm32f103 and Raspberry Pi 4B+ platforms, serves as a novel benchmark for assessing RL algorithms under conditions that closely emulate real-world scenarios.
Our website: https://ethan-chen-plus.github.io/RealWorldRLLabs
- Advanced RL and DRL algorithms for precision control of robotic angles.
- The APS environment, a custom simulation tailored for angle seeking rather than balance.
- Integration with Raspberry Pi and STM32F103ZET6 microcontroller for real-world application.
- Comprehensive experimental setups for both simulated and physical environments detailed in the paper.
- Python 3.x
- PyTorch (for specific algorithm implementations)
- Hardware: Raspberry Pi 4B+, STM32F103ZET6 microcontroller, L298N motor driver, DC3V-6V geared DC motors (TT motors), and additional components as listed in the paper's appendix.
Clone the repository:
git clone https://github.com/Ethan-Chen-plus/RoboAngleSeeker.git
Usage examples include detailed command lines for running different DQN models and configurations as presented in the research, facilitating replication of the experiments and further exploration of the algorithms' behaviors in the APS environment.
-
Default DQN Training
python main.py --model DQN
-
Double DQN with Custom Parameters
python main.py --model DoubleDQN --lr 0.005 --num_episodes 300 --epsilon 0.05
-
Dueling DQN with Increased Batch Size and Buffer Size
python main.py --model DuelingDQN --batch_size 128 --buffer_size 10000
Currently supported algorithms are listed in the table below:
Algorithm | Supported | Convergence Time |
---|---|---|
DQN | ✅ | 🐇 |
Double DQN | ✅ | 🐇 |
Dueling DQN | ✅ | 🐇 |
REINFORCE | ✅ | 🐇 |
Actor Critic | ✅ | 🐢 |
TRPO | ✅ | 🐇 |
PPO | ✅ | 🐢 |
DDPG | ✅ | 🐇 |
SAC | ✅ | 🐇 |
Behavior Clone | ✅ | 🐢 |
GAIL | ✅ | 🐇 |
PETS | ✅ | 🐢 |
MBPO | ✅ | 🐢 |
The benchmark demonstrates the effectiveness and challenges of RL and DRL algorithms in physical settings, as discussed extensively in our paper. Results provide insights into the nuances of algorithm performance across simulated and real environments.
Please cite our work if it assists in your research:
@inproceedings{Anonymous,
title={Precision Angle Seeking in Robots: A Reinforcement Learning Approach in Simulation and Reality},
author={Anonymous Author(s)},
year={2024},
booktitle={Proceedings}
}
This project is licensed under the MIT License - see the LICENSE
file for more details.
- Acknowledgments to funding bodies, research groups, and any other support should be listed here.