Change initialization strategy in DeepSoI if NLR is not fully constru…

…cted
NeuralNetworkVerification · Aug 13, 2024 · d20f839 · d20f839
1 parent a0ac1ce
commit d20f839
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/engine/SumOfInfeasibilitiesManager.cpp b/src/engine/SumOfInfeasibilitiesManager.cpp
@@ -32,6 +32,10 @@ SumOfInfeasibilitiesManager::SumOfInfeasibilitiesManager( const InputQuery &inpu
           Options::get()->getFloat( Options::PROBABILITY_DENSITY_PARAMETER ) )
     , _statistics( NULL )
 {
+    if ( !inputQuery._networkLevelReasoner ||
+         inputQuery._networkLevelReasoner->getConstraintsInTopologicalOrder().size() <
+             _plConstraints.size() )
+        _initializationStrategy = SoIInitializationStrategy::CURRENT_ASSIGNMENT;
 }
 
 void SumOfInfeasibilitiesManager::resetPhasePattern()
@@ -82,13 +86,12 @@ void SumOfInfeasibilitiesManager::initializePhasePattern()
 
     resetPhasePattern();
 
-    if ( _initializationStrategy == SoIInitializationStrategy::INPUT_ASSIGNMENT &&
-         _networkLevelReasoner )
+    if ( _initializationStrategy == SoIInitializationStrategy::INPUT_ASSIGNMENT )
     {
+        ASSERT( _networkLevelReasoner );
         initializePhasePatternWithCurrentInputAssignment();
     }
-    else if ( _initializationStrategy == SoIInitializationStrategy::CURRENT_ASSIGNMENT ||
-              !_networkLevelReasoner )
+    else if ( _initializationStrategy == SoIInitializationStrategy::CURRENT_ASSIGNMENT )
     {
         initializePhasePatternWithCurrentAssignment();
     }