Rapid Similarity Searching of Large Molecule Files

Problem

You want to perform rapid 2D similarity search on large molecule files.

Ingredients

Difficulty level

../_images/chilly.png ../_images/chilly.png

Download

Download code

makefastfp.py to generate binary fast fingerprint files

searchfastfp.py to search binary fast fingerprint files

See also the Usage (makefastfp) and Usage (searchfastfp) subsections.

Solution

In order to solve this problem two databases are utilized:

See also

This recipe discusses two code examples:

  • Generating Fingerprints section shows how to generate fingerprints and save them into a binary fingerprint file.

  • Searching Fingerprints section illustrates how to performed rapid similarity search using the pre-generated fingerprints.

Generating Fingerprints

In order to accelerate the molecule similarity search, fingerprints that encode local features of molecules are generated and stored in a binary file with a .fpbin extension. The following code snippet shows how to generate a binary fingerprint file. See Figure 1. that illustrates the overall process of the fingerprint generation.

First, an OEMolDatabase object is initialized with a input molecule filename (lines 1-3), followed by the initialization of fingerprint type from the command line interface (lines 5-6) using the OESetupFingerPrint function. A binary fingerprint file is then generated by the OECreateFastFPDatabaseFile function (line 16). The fingerprint generation is multi-thread. The number of processors that are utilized can be controlled via the OECreateFastFPDatabaseOptions class. (See also: Performance of fingerprint generation section)

 1
 2    ifname = itf.GetString("-in")
 3    ffname = itf.GetString("-fpdb")
 4
 5    if oechem.OEGetFileExtension(ffname) != "fpbin":
 6        oechem.OEThrow.Fatal("Fingerprint database file should have '.fpbin' file extension!")
 7
 8    idxfname = oechem.OEGetMolDatabaseIdxFileName(ifname)
 9
10    if not os.path.exists(idxfname):
11        if not oechem.OECreateMolDatabaseIdx(ifname):
12            oechem.OEThrow.Warning("Unable to create %s molecule index file", idxfname)
13
14    print("Using %s index molecule file" % idxfname)
15
16    moldb = oechem.OEMolDatabase()
17    if not moldb.Open(ifname):
18        oechem.OEThrow.Fatal("Cannot open molecule database file!")
19
20    fptype = oegraphsim.OESetupFingerPrint(itf)

The generated binary fingerprint file (with .fpbin extension) stores the following information for each molecule:

  • the index of the corresponding molecule in the OEMolDatabase object

  • the bitvector of the fingerprint

  • the popcount of the fingerprint i.e the number of bits that are set in the given fingerprint

../_images/makefastfp.png

Figure 1. Schematic representation of fast fingerprint generation process

Usage (makefastfp)

The first example shows how to generate a binary fingerprint file that stores tree fingerprints (that is the default type) for the eMolecules database:

  1. Retrieve eMolecules dataset in SMILES format from web server

  2. Unzip file, remove first line (header) and only keep the SMILES without title

  3. Generate tree fingerprints using makefastfp.py

Usage

makefastfp.py

1prompt > wget http://downloads.emolecules.com/free/2018-02-01/version.smi.gz
2prompt > gzip -dc version.smi.gz | awk '{if (NR!=1) {print $1 }}' > eMolecules.ism
3prompt > python3 makefastfp.py -in eMolecules.ism -fpdbfname eMolecules-tree.fpbin

Running the above commands will generate the following output:

Using eMolecules.ism.idx index molecule file
Using fingerprint type Tree,ver=2.0.0,size=4096,bonds=0-4,atype=AtmNum|Arom|Chiral|FCharge|HvyDeg|Hyb,btype=Order
Generating fingerprints with 16 threads
.....
Total: 10697014 fingerprints processed.
268.83 secs to generate 10697014 fingerprints

The generated eMolecules-tree.fpbin file stores not only fingerprints but also information about:

  • the endianess of the binary fingerprint file

  • the name of the molecule file to which the fingerprint file corresponds

  • the string representation of the fingerprint type

  • the number of fingerprints in the binary file

This information can be accessed via the OEFastFPDatabaseParams class or by peeking into the fingerprint file:

prompt > head -4 eMolecules-tree.fpbin

LE
eMolecules.ism
Tree,ver=2.0.0,size=4096,bonds=0-4,atype=AtmNum|Arom|Chiral|FCharge|HvyDeg|Hyb,btype=Order
10697014

Note

The fast fingerprint search does not support endianness compatibility for performance reason. That means that a binary fingerprint file that has been generated in a little-endian computer cannot be search in a computer that uses the big-endian encoding.

The fast fingerprint search also supports circular or path fingerprint types. The following examples show how to generate the corresponding binarry fingerprint files:

Usage

> python3 makefastfp.py -in eMolecules.ism -fpdbfname eMolecules-path.fpbin -fptype Path
> python3 makefastfp.py -in eMolecules.ism -fpdbfname eMolecules-circ.fpbin -fptype Circular

Note

GraphSim TK currently only supports popcount search method for fingerprints with the size of multiple of 256. This means that the MACCS key fingerprint type is currently not supported. When customizing other fingerprint types such as tree, circular path , the size of the fingerprint must be a multiple of 256.

See also User-defined Fingerprint section in the GraphSim TK manual.

The following example shows how to generate custom fingerprints by specifying:

  • the size of the fingerprint (using -numbits parameter)

  • the atom and bond properties that are encoded into each fingerprint (using -atomtype and -bondtype parameters, respectively)

Usage

> python3 makefastfp.py -in eMolecules.ism -fpdbfname eMolecules-custom.fpbin -fptype Tree -numbits 1024 -atomtype "AtmNum|Arom|HvyDeg|" -bondtype "Order|InRing"

Using eMolecules.ism.idx index molecule file
Using fingerprint type Tree,ver=2.0.0,size=1024,bonds=0-4,atype=AtmNum|Arom|HvyDeg,btype=Order|InRing
Generating fingerprints with 16 threads
.....

Searching Fingerprints

The following code example illustrates how to search the binary fingerprint file generated by the previous example. See Figure 2. that illustrates the overall fingerprint search process.

In this example two databases are utilized:

The correspondence between the two databases are maintained by molecule indexes (see Figure 2.). This correspondence can be checked by calling the OEAreCompatibleDatabases function (lines 16-17).

After initializing the databases, OEFPDatabaseOptions object is created (lines 24-25) that controls how the fingerprint database is searched (lines 29-31). For more details about the search options see the Fingerprint search options section The OEFastFPDatabase.GetSortedScores method returns an iterator over the calculated similarity scores (OESimScore) in sorted order. Each OESimScore holds a similarity score and index of the corresponding fingerprint of the database. This index can be used to access the original molecule in the OEMolDatabase object (lines 33-39) and write the hits into an output molecule file.

 1    # initialize databases
 2
 3    timer = oechem.OEWallTimer()
 4    timer.Start()
 5
 6    ifs = oechem.oemolistream()
 7    if not ifs.open(qfname):
 8        oechem.OEThrow.Fatal("Cannot open input file!")
 9
10    ofs = oechem.oemolostream()
11    if not ofs.open(ofname):
12        oechem.OEThrow.Fatal("Cannot open output file!")
13
14    query = oechem.OEGraphMol()
15    if not oechem.OEReadMolecule(ifs, query):
16        oechem.OEThrow.Fatal("Cannot read query molecule!")
17
18    moldb = oechem.OEMolDatabase()
19    if not moldb.Open(mfname):
20        oechem.OEThrow.Fatal("Cannot open molecule database!")
21
22    memtype = oegraphsim.OEGetFPDatabaseMemoryType(itf)
23
24    fpdb = oegraphsim.OEFastFPDatabase(ffname, memtype)
25    if not fpdb.IsValid():
26        oechem.OEThrow.Fatal("Cannot open fingerprint database!")
27    nrfps = fpdb.NumFingerPrints()
28    memtypestr = fpdb.GetMemoryTypeString()
29
30    if not oegraphsim.OEAreCompatibleDatabases(moldb, fpdb):
31        oechem.OEThrow.Fatal("Databases are not compatible!")
32
33    print("%5.2f sec to initialize databases" % timer.Elapsed())
34
35    fptype = fpdb.GetFPTypeBase()
36    print("Using fingerprint type %s" % fptype.GetFPTypeString())
37
38    opts = oegraphsim.OEFPDatabaseOptions()
39    oegraphsim.OESetupFPDatabaseOptions(opts, itf)
../_images/searchfastfp.png

Figure 2. Schematic representation of fast fingerprint search process

Usage (searchfastfp)

The following example shows how to search molecules that are similar to caffeine in the eMolecules database using the memory-mapped mode. The SMILES string of the query molecule is read from the standard input (-query .ism) and the SMILES of the top ten most similar molecules are written into standard output (-out .ism).

Usage

searchfastfp.py

prompt > python3 searchfastfp.py -fpdb eMolecules-tree.fpbin -molfname eMolecules.ism -query .ism -out .ism
c12c(n(c(n(c1=O)C)=O)C)ncn2C

Running the above command will generate the following output:

4.75 sec to initialize databases
Using fingerprint type Tree,ver=2.0.0,size=4096,bonds=0-4,atype=AtmNum|Arom|Chiral|FCharge|HvyDeg|Hyb,btype=Order
5.07 sec to search 10697015 fingerprints memory-mapped

Cn1cnc2c1c(=O)n(c(=O)n2C)N
Cn1c2c(c(=O)n(c1=O)C)n(cn2)N
Cn1cnc2c1c(=O)n(c(=O)n2C)Cn3c(=O)c4c(ncn4C)n(c3=O)C
Cn1cnc2c1c(=O)n(c(=O)n2C)CCn3c(=O)c4c(ncn4C)n(c3=O)C
Cn1cnc2c1c(=O)n(c(=O)n2C)CC#C
Cn1cnc2c1c(=O)n(c(=O)n2C)CCCO
Cn1cnc2c1c(=O)n(c(=O)n2C)CC=C
CCn1cnc2c1c(=O)n(c(=O)n2C)C
Cn1c2c(c(=O)n(c1=O)C)n(cn2)C=C
Cn1cnc2c1c(=O)n(c(=O)n2C)CCCBr

When the hits are saved into an sdf file, the similarity scores will be written to each hit with the <Similarity score> tag. The in-memory mode can be set using the -memorytype parameter.

Usage

prompt > python3 searchfastfp.py -fpdb eMolecules-tree.fpbin -molfname eMolecules.ism -query caffeine.ism -out hits.sdf -memorytype in-memory

Discussion

Performance of fingerprint generation

The graph below shows the speed of generating 1 million 4096-bit long tree fingerprints using different number of processes. The performance has been benchmarked using m4.10x large AWS instance.

../_images/makefastfp-performance.png

Graph 1. The performance of the multi-threaded fingerprint generation for 1M fingerprints

Fingerprint search options

The OEFastFPDatabase class provides several ways to search fingerprints by utilizing the OEFPDatabaseOptions class:

  • By default, OETanimoto similarity coefficient is used to quantify the degree of resemblance between two fingerprints. However, other built-in similarity coefficients such as OECosine, OEDice, OEEuclidean, OEManhattan, and OETversky can also be used (see also OEFPDatabaseOptions.SetSimFunc method).

    Note

    The current implementation of fast fingerprint search method does not allow to utilize user-defined similarity measure. If this is desired the slower fingerprint search implemented in the OEFPDatabase class should be used.

  • By default, the OEFPDatabase.GetSortedScores method returns similarity scores in descending order i.e. it identifies molecules that are most similar to the query. However, the order can be reversed to identify molecules that are most dissimilar to the query (see also OEFPDatabaseOptions.SetDescendingOrder method).

  • There is no default cut-off value defined for fingerprint search, since a reasonable cut-off value can depend on various parameters such as: the fingerprint type, the query molecule, and the the similarity measure used. The user can specify a cut-off value for a specific search using the OEFPDatabaseOptions.SetCutOff method.

  • By default, there is no limit for number of similarity scores returned, however searching and ordering millions of molecules are not recommended. A reasonable limit can be set by using the OEFPDatabaseOptions.SetLimit method.

See also in OEChem TK manual

Theory

API

See also in GraphSim TK manual

Theory

API