1 /* Copyright (C) 2006-2008 Joris Mooij [joris dot mooij at tuebingen dot mpg dot de]
2 Radboud University Nijmegen, The Netherlands /
3 Max Planck Institute for Biological Cybernetics, Germany
5 This file is part of libDAI.
7 libDAI is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 libDAI is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with libDAI; if not, write to the Free Software
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
24 /// \brief Defines class HAK.
27 #ifndef __defined_libdai_hak_h
28 #define __defined_libdai_hak_h
32 #include <dai/daialg.h>
33 #include <dai/regiongraph.h>
35 #include <dai/properties.h>
41 /// Approximate inference algorithm: implementation of single-loop ("Generalized Belief Propagation") and double-loop algorithms by Heskes, Albers and Kappen
42 class HAK
: public DAIAlgRG
{
44 std::vector
<Factor
> _Qa
;
45 std::vector
<Factor
> _Qb
;
46 std::vector
<std::vector
<Factor
> > _muab
;
47 std::vector
<std::vector
<Factor
> > _muba
;
48 /// Maximum difference encountered so far
50 /// Number of iterations needed
54 /// Parameters of this inference algorithm
56 /// Enumeration of possible cluster choices
57 DAI_ENUM(ClustersType
,MIN
,DELTA
,LOOP
)
62 /// Maximum number of iterations
71 /// How to choose the clusters
72 ClustersType clusters
;
74 /// Use single-loop (GBP) or double-loop (HAK)
77 /// Depth of loops (only relevant for clusters == ClustersType::LOOP)
81 /// Name of this inference algorithm
82 static const char *Name
;
85 /// Default constructor
86 HAK() : DAIAlgRG(), _Qa(), _Qb(), _muab(), _muba(), _maxdiff(0.0), _iters(0U), props() {}
89 HAK( const HAK
&x
) : DAIAlgRG(x
), _Qa(x
._Qa
), _Qb(x
._Qb
), _muab(x
._muab
), _muba(x
._muba
), _maxdiff(x
._maxdiff
), _iters(x
._iters
), props(x
.props
) {}
91 /// Assignment operator
92 HAK
& operator=( const HAK
&x
) {
94 DAIAlgRG::operator=( x
);
99 _maxdiff
= x
._maxdiff
;
106 /// Construct from FactorGraph fg and PropertySet opts
107 HAK( const FactorGraph
&fg
, const PropertySet
&opts
);
109 /// Construct from RegionGraph rg and PropertySet opts
110 HAK( const RegionGraph
&rg
, const PropertySet
&opts
);
113 /// @name General InfAlg interface
115 virtual HAK
* clone() const { return new HAK(*this); }
116 virtual HAK
* create() const { return new HAK(); }
117 virtual std::string
identify() const;
118 virtual Factor
belief( const Var
&n
) const;
119 virtual Factor
belief( const VarSet
&ns
) const;
120 virtual std::vector
<Factor
> beliefs() const;
121 virtual Real
logZ() const;
123 virtual void init( const VarSet
&ns
);
124 virtual double run();
125 virtual double maxDiff() const { return _maxdiff
; }
126 virtual size_t Iterations() const { return _iters
; }
130 /// @name Additional interface specific for HAK
132 Factor
& muab( size_t alpha
, size_t _beta
) { return _muab
[alpha
][_beta
]; }
133 Factor
& muba( size_t alpha
, size_t _beta
) { return _muba
[alpha
][_beta
]; }
134 const Factor
& Qa( size_t alpha
) const { return _Qa
[alpha
]; };
135 const Factor
& Qb( size_t beta
) const { return _Qb
[beta
]; };
138 double doDoubleLoop();
142 void constructMessages();
143 void findLoopClusters( const FactorGraph
&fg
, std::set
<VarSet
> &allcl
, VarSet newcl
, const Var
& root
, size_t length
, VarSet vars
);
145 void setProperties( const PropertySet
&opts
);
146 PropertySet
getProperties() const;
147 std::string
printProperties() const;
151 } // end of namespace dai