Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement BaBSR Branching Heuristic as a new Branching Strategy #851

Open
wants to merge 18 commits into
base: master
Choose a base branch
from

Conversation

liamjdavis
Copy link
Contributor

This pull request introduces a new branching strategy, the BaBSR branching heuristic. The new change for this new heuristic includes additions to the configuration, option parsing, and engine components.

Key changes include:

Branching Strategy Implementation:

  • Implemented the pickSplitPLConstraintBasedOnBaBsrHeuristic method in the Engine class to split on the ReLUs with the highest BaBSR score. [1] [2] [3]
  • Added support for BaBSR heuristic in the ReluConstraint class, including methods to compute the BaBSR score. [1] [2] [3] [4] [5] [6] [7] [8] [9]

Option Parsing:

  • Updated OptionParser to include babsr-heuristic as a valid branching strategy.
  • Modified Options::getDivideStrategy() to handle the new babsr-heuristic strategy.

Testing:

  • Added a new unit test file for Test_BaBsrSplitting.h for unit tests of the branching strategy.

@MatthewDaggitt
Copy link
Collaborator

I think the new option should be documented in the CHANGELOG.md file?

@liamjdavis
Copy link
Contributor Author

liamjdavis commented Dec 17, 2024

Fixed!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants