11 #include <tracking/trackFindingVXD/trackSetEvaluator/HopfieldNetwork.h>
13 #include <framework/logging/Logger.h>
15 #include <Eigen/Dense>
22 std::vector<OverlapResolverNodeInfo>& overlapResolverNodeInfos,
unsigned short nIterations)
30 if (overlapResolverNodeInfos.size() < 2) {
31 B2DEBUG(20,
"No reason to doHopfield with less than 2 nodes!");
35 const float compatibilityValue = (1.0 -
m_omega) /
static_cast<float>(overlapResolverNodeInfos.size() - 1);
37 const size_t overlapSize = overlapResolverNodeInfos.size();
40 Eigen::MatrixXd W(overlapSize, overlapSize);
42 W.fill(compatibilityValue);
45 for (
const auto& aTC : overlapResolverNodeInfos) {
46 for (
unsigned int overlapIndex : aTC.overlaps) {
47 W(aTC.trackIndex, overlapIndex) = -1.0;
53 Eigen::VectorXd x(overlapSize);
55 for (
unsigned int i = 0; i < overlapSize; i++) {
56 x(i) = gRandom->Uniform(1.0);
60 Eigen::VectorXd xOld(overlapSize);
63 std::vector<unsigned short> sequenceVector(overlapSize);
65 std::iota(sequenceVector.begin(), sequenceVector.end(), 0);
68 B2DEBUG(100,
"sequenceVector with length " << sequenceVector.size());
69 B2DEBUG(100,
"Entries are from begin to end:");
70 for (
auto && entry : sequenceVector) {
71 B2DEBUG(100, std::to_string(entry) +
", ");
75 std::vector<float> cValues(nIterations);
80 unsigned iIterations = 0;
86 std::shuffle(sequenceVector.begin(), sequenceVector.end(),
TRandomWrapper());
90 for (
unsigned int i : sequenceVector) {
91 float aTempVal = W.row(i).dot(x);
92 float act = aTempVal +
m_omega * overlapResolverNodeInfos[i].qualityIndicator;
93 x(i) = 0.5 * (1. + tanh(act / T));
99 c = (x - xOld).cwiseAbs().maxCoeff();
100 B2DEBUG(10,
"c value is " << c <<
" at iteration " << iIterations);
101 cValues.at(iIterations) = c;
103 if (iIterations + 1 == nIterations) {
104 B2INFO(
"Hopfield reached maximum number of iterations without convergence. cValues are:");
105 for (
auto && entry : cValues) {
106 B2INFO(std::to_string(entry));
114 for (
unsigned int i = 0; i < overlapSize; i++) {
115 overlapResolverNodeInfos[i].activityState = x(i);