Figure 4a (Lower Bounds):
python -m nn_partition.example \
--partitioner GreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition lower_bnds \
--model random_weights \
--activation relu \
--show_input --show_output
Figure 4b (Linf Ball):
python -m nn_partition.example \
--partitioner GreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition linf \
--model random_weights \
--activation relu \
--show_input --show_output
Figure 4c (Convex Hull):
python -m nn_partition.example \
--partitioner GreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model random_weights \
--activation relu \
--show_input --show_output
Fig 4a | Fig 4b | Fig 4c |
---|---|---|
Lower Bounds | Linf Ball | Convex Hull |
Figure 5a (SG+IBP):
python -m nn_partition.example \
--partitioner SimGuided \
--propagator IBP_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model random_weights \
--activation relu \
--input_plot_labels None None \
--show_input --skip_show_output \
--input_plot_aspect equal
Figure 5b (SG+CROWN):
python -m nn_partition.example \
--partitioner SimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model random_weights \
--activation relu \
--input_plot_labels None None \
--show_input --skip_show_output \
--input_plot_aspect equal
Figure 5c (GSG+CROWN):
python -m nn_partition.example \
--partitioner GreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model random_weights \
--activation relu \
--input_plot_labels None None \
--show_input --skip_show_output \
--input_plot_aspect equal
Figure 5d (GSG+CROWN):
python -m nn_partition.example \
--partitioner AdaptiveGreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model random_weights \
--activation relu \
--input_plot_labels None None \
--show_input --skip_show_output \
--input_plot_aspect equal
Fig 5a | Fig 5b | Fig 5c | Fig 5d |
---|---|---|---|
SG+IBP | SG+CROWN | GSG+IBP | GSG+CROWN |
Figure 6a (SG+IBP):
python -m nn_partition.example \
--partitioner SimGuided \
--propagator IBP_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model robot_arm \
--activation tanh \
--output_plot_labels x y \
--output_plot_aspect equal \
--skip_show_input
Figure 6b (AGSG+CROWN):
python -m nn_partition.example \
--partitioner AdaptiveGreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 2 \
--interior_condition convex_hull \
--model robot_arm \
--activation tanh \
--output_plot_labels x y \
--output_plot_aspect equal \
--skip_show_input
Fig 6a | Fig 6b |
---|---|
SG+IBP | AGSG+CROWN |
This figure unfortunately requires code for the RL implementation that is under IP protection from our research sponsor.
Info coming soon...
For any of the above examples, you can add the --make_animation
flag which will save a .gif
in results/animations/
, e.g.,
python -m nn_partition.example \
--partitioner GreedySimGuided \
--propagator CROWN_LIRPA \
--term_type time_budget \
--term_val 5 \
--interior_condition lower_bnds \
--model random_weights \
--activation relu \
--show_input --show_output \
--make_animation