1. Field of Invention
This invention relates to iterative decoding of input sequences.
2. Description of Related Art
Maximum a posteriori (MAP) sequence decoding selects a most probable information sequence X1T=(X1, X2, . . . , XT) that produced the received sequence Y1T=(Y1, Y2, . . . , YT). For transmitters and/or channels that are modeled using Hidden Markov Models (HMM), the process for obtaining the information sequence X1T that corresponds to a maximum probability is difficult due to a large number of possible hidden states as well as a large number of possible information sequences X1T. Thus, new technology is needed to improve MAP decoding for HMMs.
This invention provides an iterative process to maximum a posteriori (MAP) decoding. The iterative process uses an auxiliary function which is defined in terms of a complete data probability distribution. The MAP decoding is based on an expectation maximization (EM) algorithm which finds the maximum by iteratively maximizing the auxiliary function. For a special case of trellis coded modulation, the auxiliary function may be maximized by a combination of forward-backward and Viterbi algorithms. The iterative process converges monotonically and thus improves the performance of any decoding algorithm.
The MAP decoding decodes received inputs by minimizing a probability of error. A direct approach to achieve this minimization results in a complexity which grows exponentially with T, where T is the size of the input. The iterative process avoids this complexity by converging on the MAP solution through repeated use of the auxiliary function.
The invention is described in detail with reference to the following figures where like numerals reference like elements, and wherein:
Maximum a posteriori (MAP) sequence decoding is a technique that decodes the received sequence Y1T by minimizing a probability of error to obtain X1T (and if a model of the transmitter 102 is included, to obtain I1T). In MAP, the goal is to choose a most probable X1T that produces the received Y1T The MAP estimator may be expressed by equation 1 below.
where Pr(·) denotes a corresponding probability or probability density function and {circumflex over (X)}1T is an estimate of X1T. Equation 1 sets {circumflex over (X)}1T to the X1T that maximizes Pr(X1T, Y1T).
The Pr (X1T, Y1T) term may be obtained by modeling the channel 106 of the communication system 100 using techniques such as Hidden Markov Models (HMMs). An input-output HMM λ=(S,X,Y,π,{P(X,Y)}) is defined by its internal states S={1,2, . . . n}, inputs X, outputs Y, initial state probability vector λ, and the input-output probability density matrices (PDMs) P(X,Y), XεX, YεY. The elements of P(X,Y), pij(X,Y)=Pr(j,X,Y|i), are conditional probability density functions (PDFs) of input x and corresponding output y after transferring from the state i to state j. It is assumed that the state sequence S0t=(S0, S1, . . . , St), input sequence X1t=(X1, X2, . . . Xt), and output sequence Y1t=(Y1, Y2, . . . , Yt) possess the following Markovian property
Pr(St,Xt,Yt|S0t−1,Y1t−1)=Pr(St,Xt,Yt|St−1).
Using HMM, the PDF of the input sequence X1T and output sequence Y1T may be expressed by equation 2 below:
where 1 is a column vector of n ones, π is a vector of state initial probabilities, and n is a number of states in the HMM. Thus, the MAP estimator when using HMM may be expressed by equation 3 below:
The maximization required by equation 3 is a difficult problem because all possible sequences of X1T must be considered. This requirement results in a complexity that grows exponentially with T. This invention provides an iterative process to obtain the maximum without the complexity of directly achieving the maximization by evaluating equation 2 for all possible X1T, for example. In the iterative process, an auxiliary function is developed whose iterative maximization generates a sequence of estimates for X1T approaching the maximum point of equation 2.
The iterative process is derived based on the expectation maximization (EM) algorithm. Because the EM algorithm converges monotonically, the iterative process may improve the performance of any decoding algorithm by using its output as an initial sequence of the iterative decoding algorithm. In the following description, it is assumed that HMM parameters for the channel 106 and/or the transmitter 102 are available either by design or by techniques such as training.
The auxiliary function may be defined in terms of a complete data probability distribution shown in equation 4 below.
where z=i0T is an HMM state sequence, πi
where p is a number of iterations and Q(Xit, Xi,pt) is the auxiliary function which may be expressed as
The auxiliary function may be expanded based on equation 4 as follows:
where C does not depend on X1T, n is a number of states in the HMM and
γt,ij(X1,pT)=αi(X1,pt−1,Y1t−1)pij(Xt,p,Yt)βj(Xt+1,pT,Y1+1T) (8)
where αi(X1,pt, Y1t) and βj(Xt+1,pT, Yt+1T) are the elements of the following forward and backward probability vectors
Based on equations 5-9, the iterative process may proceed as follows. At p=0, an initial estimate of X1,0T is generated. Then, Q(X1T,X1,0T) is generated for all possible sequences of X1T. From equations 7 and 8, Q(X1T,X1,0T) may be evaluated by generating γt,ij(X1,0T) and log (pij(Xt,Yt)) for each t, i, and j. γt,ij(X1,0T) may be generated by using the forward-backward algorithm as shown below:
α(X1,p0,Y10)=π, α(X1,pt,Y1t)=α(X1,pt−1,Y1t−1)P(Xt,p,Yt), t=1,2, . . . T
β(XT+1,pT,YT+1T)=1, β(Xt,pT,YtT)=P(Xt+1,pT,Yt+1T), t=T−1,T−2, . . . , 1
Log(pij(Xt,Yt)) is generated for all possible Xt for t=1, 2, . . . , T and the (Xt)s that maximize D(X1T,X1,0T) are selected as X1,1T. After X1,1T is obtained, it is compared with X1,0T. If a measure D(X1,1T,X1,0T) of difference between the sequences exceeds a compare threshold, then the above process is repeated until the difference measure D(X1,pT,X1,p−1T) is within the threshold. The last X1,pT for p iterations is the decoded output. The measure of difference may be an amount of mismatch information. For example, if X1T is a sequence of symbols, then the measure may be a number of different symbols between X1,pT and X1,p−1T (Hamming distance); if X1T is a sequence of real numbers, then the measure may be an Euclidean distance D(X1,pT,X1,p−1T)=[Σi=1T(Xi,p−Xi,p−1)2]1/2.
In step 1008, the receiver 104 selects a sequence X1,p+1T that maximizes Q(X1,p+1T,X1,pT) and goes to step 1010. In step 1010, the receiver 104 compares X1,pT with X1,p+1T. If the compare result is within the compare threshold, then the receiver 104 goes to step 1012; otherwise, the receiver 104 returns to step 1004 and continues the process with the new sequence X1,p+1T. In step 1012, the receiver 104 outputs X1,p+1T, and goes to step 1014 and ends the process.
The efficiency of the above described iterative technique may be improved if the transmitted sequence is generated by modulators such as a trellis coded modulator (TCM). A TCM may be described as a finite state machine that may be defined by equations 10 and 11 shown below.
S
t+1
=f
t(St,It) (10)
X
t
=g
t(St,It) (11)
Equation 10 specifies the TCM state transitions while equation 11 specifies the transmitted information sequence based on the state and the input information sequence. For example, after receiving input It in state St, the finite state machine transfers to state St+1, based on St and It as shown in equation 10. The actual output by the transmitter 102 is Xt according to equation 11. Equation 10 may represent a convolutional encoder and equation 11 may represent a modulator. For the above example, the transmitter output information sequence X1T may not be independent even if the input information sequence I1T is independent.
In equation 15, the log(pij(Yt,Xt)) term may be analyzed based on the TCM state transitions because the information actually transmitted X is related to the source information It by Xt=gt(St, It). This relationship between Xt and It forces many elements pij(Yt,Xt) of P (Yt,Xt), to zero since the finite state machine (equations 10 and 11) removes many possibilities that otherwise must be considered. Thus, unlike the general case discussed in relation to equations 5-9, evaluation of pij(Yt,Xt) may be divided into a portion that is channel related and another portion that is TCM related. The following discussion describes the iterative technique in detail for the TCM example.
For a TCM system with an independent and identically distributed information sequence, an input-output HMM may be described by equations 12 and 13 below.
Pc(Yt|Xt) is the conditional PDM of receiving Yt given that Xt has been transmitted for the HMM of a medium (channel) through which the information sequence is transmitted; ps
where π is a vector of the initial probabilities of the channel states, Xt=gt(St,It), and the product is taken along the state trajectory St+1=ft(St,It) for t=1, 2, . . . , T.
If all elements of the input information sequence are equally probable, then the MAP estimate may be expressed by equation 14 below.
The auxiliary function may be expressed by equations 15-17 below corresponding to equations 7-9 above.
where Xt=gt(St,It) and
γt,ij(I1,pT)=αi(Y1t−1|T1,p)pc,ij(Yt|Xt,p)βj(Yt+1,pT|It+1,pT) (16)
αi(Y1T−1|I1,pt−1) and βj(Yt+1,pT|It+1,pT) are the elements of the forward and backward probability vectors
From equation 15, the Viterbi algorithm may be applied with the branch metric
to find a maximum of Q(I1T, I1,pT) which can be interpreted as a longest path leading from the initial zero state to one of the states ST where only the encoder trellis is considered. The Viterbi algorithm may be combined with the backward portion of the forward-backward algorithm as follows.
1. Select an initial source information sequence I1,0T=I1,0, I2,0, . . . , IT,0
2. Forward part:
α(Y1t|T1,pt)=α(Y1t−1|I1,pt−1)Pc(Yt|X1,p),
3. Backward part:
This step selects the paths with the largest lengths (the survivors).
This step estimates It corresponding to the state St by selecting the It of the survivor in step d.
4. Reestimate the information sequence:
I
t,p+1
=Î
t(Ŝt+1=ft(Ŝt,It,p+1), t=1,2, . . . , T where Ŝ1=0; and
5. If Tt,p+1≠It,p, go to step 2; otherwise decode the information sequence as It,p+1T.
The Viterbi algorithm together with the backward algorithm may be used to obtain a next estimate of the input information sequence I1,15. This process begins with the state transitions between t=4 and t=5 by selecting state transitions leading to each of the states s0-s3 at t=4 from states at t=5 that have the largest value of the branch metric L(S4)=m(I5) of equation 18 above. Then, the process moves to select state transitions between the states at t=3 and t=4 that have the largest cumulative distance L(S3)=L(S4)+m(I4). This process continues until t=0 and the sequence of input information I′ corresponding to the path connecting the states from t=0 to t=5 that has the longest path
is selected as the next input information sequence I1,15.
For the example in
The branch metric m(It) of equation 18 represents a “distance” between the states and is used to select the state transition that corresponds to the longest path for each of the states s0-s3 at t=4:
where βj(X6,05, Y65)=1, and X5=g5(S5, I5) by definition. There is an I5 that corresponds to each of the state transitions 301-308. For this example, L(S4)=m(I5) corresponding to odd numbered state transitions 301-307 are greater than that for even numbered state transitions 302-308. Thus, odd numbered state transitions are “survivors.” Each of them may be part of the state trajectory that has the longest path from t=0 to t=5. This transition (the survivor) is depicted by the solid arrow while the transitions with smaller lengths are depicted by dashed lines.
The state sequence determination process continues by extending the survivors to t=3 as shown in
For this example, the distances corresponding to the odd numbered state transitions 309-315 are longer than distances corresponding to even numbered state transitions 310-316. Thus, the paths corresponding to the odd numbered state transitions are the survivors. As shown in
The above process continues until t=0 is reached as shown in
All the functions of the forward, backward and maximal length processors 206, 208 and 210 may also be performed by the controller 202 which may be either a general purpose or special purpose computer (e.g., DSP).
The forward processor 206 generates the forward probability vectors αi(X1,pt−1,Y1t−1) herein referred to as αi. For every iteration, when a new X1,pT (or I1,pT) is generated, the forward processor 206 may generate a complete set of αi.
The backward processor 208 together with the maximal length processor 210 generate a new state sequence by searching for maximal length state transitions based on the branch metric m(It). Starting with the final state transition between states corresponding to t=T−1 and t=T, the backward processor generates βj(Xt+1,pT, Yt+1T) (hereinafter referred as βj) as shown in equation 8 for each state transition.
The maximal length processor 210 generates m(It) based on the results of the forward processor 206, the backward processor 208 and pij(Xt,Yt). After generating all the m(It)s corresponding to each of the possible state transitions, the maximal length processor 210 compares all the L(St)+m(It)s and selects the state transition that corresponds to the largest L(St)+m(It), and the It (via St+1=ft(St, It)) that corresponds to the selected state transition is selected as the estimated input information for that t. The above process is performed for each t=1, 2, . . . , T to generate a new estimate I1,pT for each of the iteration p.
Initially, the controller 202 places an estimate of the PDM P(X,Y) and π in the memory 204 that corresponds to the HMM for the channel 106 and/or the transmitter 102. The PDM P(X,Y) may be obtained via well known training processes, for example.
When ready, the controller 202 receives the received input information sequence Y1T and places them in the memory 204 and selects an initial estimate of I1,0T (or X1,0T). The controller 202 coordinates the above-described iterative process until a new estimate I1,1T (or X1,1T) is obtained. Then, the controller 202 compares I1,0T with I1,1T to determine if the compare result is below the compare threshold value (e.g., matching a predetermined number of elements or symbols of the information sequence). The compare threshold may be set to 0, in which case I1,0T must be identical with I1,1T. If an acceptable compare result is reached, I1,1T is output as the decoded output. Otherwise, the controller 202 iterates the above-described process again and compares the estimated I1,pT with I1,pT until an acceptable result is reached and I1,pT is output as the decoded output.
In step 2006, the controller 202 decrements t and goes to step 2008. In step 2008, the controller 202 determines whether t is equal to 0. If t is equal to 0, the controller 202 goes to step 2010; otherwise, the controller 202 returns to step 2004. In step 2010, the controller 202 outputs the new estimated ItT and goes to step 2012 and returns to step 1006 of
A specific example of the iterative process for convolutional encoders is enclosed in the appendix.
While this invention has been described in conjunction with specific embodiments thereof, it is evident that many alternatives, modifications, and variations will be apparent to those skilled in the art. Accordingly, preferred embodiments of the invention as set forth herein are intended to be illustrative, not limiting. Various changes may be made without departing from the spirit and scope of the invention.
For example, a channel may be modeled as Pc(Y|X)=PcBc(Y|X) where Pc is a channel state transition probability matrix and Bc(Y|X) is a diagonal matrix of state output probabilities. For example, based on the Gilbert-Elliott model
where
This is a continuation of U.S. patent application Ser. No. 10/789,794 filed Feb. 28, 2004, which is a continuation of U.S. patent application Ser. No. 09/549,201 filed Apr. 13, 2000, that followed provisional application No. 60/174,601 entitled “Map Decoding In Channels With Memory” filed on Jan. 5, 2000 now U.S. Pat. No. 6,700,937, issued Mar. 2, 2004. All materials included in the aforementioned applications directly or by incorporation are hereby incorporated by reference including all references cited therein.
Number | Date | Country | |
---|---|---|---|
60174601 | Jan 2000 | US |
Number | Date | Country | |
---|---|---|---|
Parent | 10789794 | Feb 2004 | US |
Child | 12290811 | US | |
Parent | 09549201 | Apr 2000 | US |
Child | 10789794 | US |