neuro_symbolic_datalog_bridge.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "pulsatrix/datalog_dual_semiring.hpp"
#include "pulsatrix/datalog_lrp.hpp"
#include "pulsatrix/datalog_rule.hpp"
#include "pulsatrix/datalog_semiring.hpp"
#include "pulsatrix/datalog_weighted_fact_database.hpp"
#include "pulsatrix/linear_module.hpp"
#include "pulsatrix/lrp_rule_config.hpp"
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix::datalog {
 
struct NeuralPredicateQueryResult {
    double query_weight;             
    double grad_wrt_predicate_output;  
};
 
struct NeuralPredicateRelevanceResult {
    RelevanceMap base_fact_relevance;
    Tensor relevance_wrt_x;
};
 
class NeuralPredicateDatalogBridge {
public:
    explicit NeuralPredicateDatalogBridge(DeviceBackend* backend);
 
    [[nodiscard]] NeuralPredicateQueryResult evaluate(const Tensor& x);
 
    void backward();
 
    [[nodiscard]] NeuralPredicateRelevanceResult propagate_relevance(double relevance_seed,
                                                                       const LRPRuleConfig& config = LRPRuleConfig{});
 
    [[nodiscard]] LinearModule& predicate() { return predicate_; }
 
    [[nodiscard]] static std::vector<Rule> diamond_ancestor_program();
 
    [[nodiscard]] static WeightedFactDatabase<double> constant_edge_facts();
 
private:
    DeviceBackend* backend_;
    LinearModule predicate_;
    Tensor last_x_;
    Tensor last_s_;  // (1,1) -- cached sigmoid output, sigmoid_backward's own "y" input.
    double last_grad_wrt_s_ = 0.0;
    bool has_evaluated_ = false;
};
 
}  // namespace pulsatrix::datalog